题目中给出了例子:例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列(如下图所示,画的有点乱。。。),但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
根据题意知道,就是判断第二个序列(B)是否为第一个序列(A)的弹栈顺序,这里考虑利用一个栈的数据结构按顺序来保存压栈的序列,即对A序列进行遍历,将每个元素保存到栈结构中,再判断栈顶元素是否与弹栈序列的元素相同,最终如果栈中元素为空则证明B是A的一个弹出顺序,反之亦反。那么在这里如果判断栈顶元素是否与弹栈序列的元素相同呢?需要通过设置一个变量来得到B序列的索引值(假设为popIndex),初始化为0,当 栈顶元素 和 弹栈序列索引值为popIndex的元素 相等时,先移除栈顶元素,再将索引值++,具体的代码如下:
public boolean IsPopOrder(int[] pushA, int[] popA) {
if (pushA.length == 0 || popA.length == 0) {
return false;
}
Stack<Integer> stack = new Stack<>();
int popIndex = 0;
for (int i = 0; i < pushA.length; i++) {
stack.push(pushA[i]);
while (!stack.isEmpty() && stack.peek() == popA[popIndex]) {
stack.pop();
popIndex++;
}
}
return stack.isEmpty();
}