题目
栈的压入、弹出序列
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1、2、3、4、5是某栈的压栈序列,判断序列4、5、3、2、1是否是该压栈序列对应的一个弹出序列?
思路
建立一个辅助栈,把输入的第一个序列中的数字依次压入该辅助栈,并按照第二个序列的顺序依次从该栈中弹出数字。
以弹出序列4、5、3、2、1为例分析压栈和弹出的过程。弹出序列中第一个要弹出的是4,那么需要将4压入栈,而从第一个序列中不难发现,要想将4压入栈中,需要先把1、2、3压入栈中,此时栈中有四个数字,从栈顶到栈底分别为4、3、2、1。弹出4后,栈中的数字为3、2、1.接下来需要弹出的数字是5,而此时栈顶不是5,就需要继续向栈中压入数据,从序列一发现,下一个要压入的数字是5,压入5后栈中的数字是5、3、2、1.弹出5后为3、2、1。不难看出此时要弹出的序列和栈中的元素的顺序是一致的。故4、5、3、2、1是该压栈序列(1、2、3、4、5)对应的一个弹出序列。整个过程如下表:
总结
判断一个序列是不是栈的弹出序列的规律是:如果下一个弹出的数字刚好是栈顶数字,那么直接弹出。如果下一个弹出的数字不再栈顶,我们就把压栈序列中没有入栈的数字压入栈中,直到把下一个需要弹出的数字压入栈为止。如果所有的数字都压入栈中了仍然没有找到下一个弹出的数字,那么此序列不可能成为一个弹出序列。
代码实现
public static boolean isPopOrder(int[] orig, int[] des) {
if (orig==null||des==null||orig.length != des.length) {
return false;
}
Stack<Integer> stack = new Stack<Integer>();
int j = 0;
int i = 0;
for (; i < des.length; i++) {
int popValue = des[i];
if (stack.isEmpty()) {
stack.push(orig[j++]);
}
int value = stack.peek();
if (popValue == value) {
stack.pop();
} else {
for (; j < orig.length; j++) {
if (popValue == orig[j]) {
j++;
break;
} else {
stack.push(orig[j]);
}
}
if (j >= orig.length) {
i++;
break;
}
}
}
for (; i < des.length && !stack.isEmpty(); i++) {
if (des[i] != stack.pop()) {
return false;
}
}
return true;
}