问题
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
(注意:这两个序列的长度是相等的)
解法
我们可以用一个辅助栈去模拟进栈出栈,我们每压入一个数,用栈顶元素与出栈数组 B[i] 去比较,如果相等,栈顶元素弹出,
i++,再比较,直到栈顶元素与 B[i] 不相等,就再往栈中压入一个数。当所有数都压入栈之后,如果出入栈的顺序是相同的话,最后辅助栈应该是空的;
public static boolean IsPopOrder(int [] pushA,int [] popA) {
if( pushA == null || popA == null || pushA.length != popA.length) return false;
int i = 0;
int j = 0;
Stack<Integer> stack = new Stack<>();
while(i < pushA.length) {
stack.push(pushA[i++]);
while(!stack.isEmpty() && stack.peek() == popA[j]) {
stack.pop();
j++;
}
}
return stack.isEmpty();
}