劍指offer(二十一)——棧的壓入、彈出序列
題目描述
輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否可能爲該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序,序列4,5,3,2,1是該壓棧序列對應的一個彈出序列,但4,3,5,1,2就不可能是該壓棧序列的彈出序列。(注意:這兩個序列的長度是相等的)
題解
我的思路是去模擬彈出序列,如果模擬成功就返回true,模擬不成功就返回false。
- 遍歷popA數組,當該元素不等於棧頂元素或棧無元素時,就不斷將pushA數組元素壓入棧中,直至棧頂元素等於該元素。
- 然後彈出棧頂元素,繼續遍歷下一個popA數組元素。
- 假如能夠成功遍歷完popA數組,則證明模擬成功。
- 假如當pushA遍歷完之後,popA還沒遍歷完,則模擬不成功。
import java.util.Stack;
public static boolean IsPopOrder(int [] pushA,int [] popA) {
Stack<Integer> tempStack = new Stack<>();
for(int i = 0, j = 0; i < popA.length; i++) {
while(tempStack.size() == 0 || popA[i] != tempStack.peek()) {
if (j >= pushA.length) {
return false;
}
tempStack.push(pushA[j]);
j++;
}
tempStack.pop();
}
return true;
}