解题思路
由于出栈序列会存在很多种可能,例如入栈顺序为1,2,3,4,5
,可能的出栈序列为1,2,3,4,5
,4,5,3,2,1
等等,不可能将所有序列进行穷举,因此需要按照入栈序列和出栈序列进行模拟,看能否在给定入栈序列下得到出栈序列,可以则成功,否则失败。
通过两个指针分别遍历入栈序列pushA
和出栈序列popA
,如果栈顶元素等于当前出栈元素,则模拟出栈,popA
后移一位,否则继续入栈,pushA
后移一位,若最终栈为空则表示成功,否则失败。
public boolean IsPopOrder(int[] pushA, int[] popA) {
Stack<Integer> stack = new Stack<>();
int index_pop = 0;
for (int index_push = 0; index_push < pushA.length; index_push++) {
stack.push(pushA[index_push]); // 入栈
while (!stack.isEmpty() && stack.peek() == popA[index_pop]) {
// 栈不空并且栈顶元素与当前压出序列元素相等,则可以出栈
stack.pop();
index_pop++;
}
}
return stack.isEmpty();
}
上面的方法需要开辟一个栈的空间进行模拟,可以优化空间复杂度。可以利用pushA
数组,因为当pushA
中的值已经入栈后,这些空间就没用了,可以用来作为栈使用。此时只需要维护一个top指针,作为栈顶指针,将上面代码段中用到stack的地方改成pushA[top]等相关操作即可
public boolean IsPopOrder(int[] pushA, int[] popA) {
int top = 0; // 栈顶指针
int index_pop = 0;
for (int index_push = 0; index_push < pushA.length; index_push++) {
pushA[top++] = pushA[index_push];
while (top > 0 && pushA[top - 1] == popA[index_pop]) {
// 栈不空并且栈顶元素与当前压出序列元素相等,则可以出栈
top--;
index_pop++;
}
}
return top == 0 ? true : false;
}