輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否爲該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序,序列4,5,3,2,1是該壓棧序列對應的一個彈出序列,但4,3,5,1,2就不可能是該壓棧序列的彈出序列。(注意:這兩個序列的長度是相等的)
思路如下:
建立一個輔助棧;
將第一個序列的數字壓入輔助棧;
如果第二個序列的下一個彈出數字剛好是棧頂數字,則直接彈出,第二個序列彈出當前數字,輔助棧也彈出該數字;
否則,就把第一個序列中尚未入棧的數字壓入輔助棧,直到把第二個序列中下一個需要彈出的數字壓入棧頂爲止。
一次遍歷之後如果棧內還有數值,則出棧順序應該和popA的順序一致,如果一次遍歷後棧空,則正確
public boolean IsPopOrder(int [] pushA,int [] popA) {
boolean ok = false;
if (pushA!=null&&popA!=null){
Stack<Integer> stack = new Stack<>();
int i=0;int j=0;
while (i<pushA.length){
if (pushA[i] !=popA[j]){
stack.push(pushA[i]);i++;
}else {
i++;j++;continue;
}
}
while (j<popA.length){
int top =0;
if ((top=stack.pop())!=popA[j]){
stack.push(top);
break;
}
else {
j++;
}
}
if (stack.isEmpty()){
ok = true;
}
}
return ok;
}