参考自:《剑指Offer——名企面试官精讲典型编程题》
题目:栈的压入与弹出序列
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
例如:序列{1、2、3、4、5}是某栈的压栈序列,序列{4、5、3、2、1}是该压栈序列对应的一个弹出序列。但{4、3、5、1、2}就不可能是该压栈序列的弹出序列。
主要思路:借助辅助栈来模拟出栈过程
,并且使用一个指针指向弹出序列当前要弹出的元素。当每次把压栈序列的元素压入辅助栈中时,
若辅助栈栈顶元素等于指针指向的元素,则弹出栈顶元素,并且将指针指向弹出序列的下一个元素,循环该过程,直到栈顶元素不等于指针指向的元素。
最后,若压栈序列遍历完,同时辅助栈也为空(或者弹出序列也遍历完),则说明该弹出序列满足条件。
关键点:辅助栈,模拟压栈和弹出过程。
时间复杂度:O(n)
public class PopOrder {
public static void main(String[] args) {
int[] pushA = {1, 2, 3, 4, 5};
int[] popA = {4, 5, 3, 2, 1};
int[] popB = {4, 3, 5, 1, 2};
//true
System.out.println(isPopOrder(pushA, popA));
//false
System.out.println(isPopOrder(pushA, popB));
}
private static boolean isPopOrder(int[] pushA, int[] popA) {
if (pushA.length == 0 || popA.length == 0) {
return false;
}
Stack<Integer> helpStack = new Stack<>();
//当前要弹出的元素位置
int popIndex = 0;
for (int a : pushA) {
helpStack.push(a);
//辅助栈栈顶元素等于弹出序列当前要弹出的元素
while (!helpStack.isEmpty() && popIndex < popA.length
&& helpStack.peek() == popA[popIndex]) {
helpStack.pop();
popIndex++;
}
}
return helpStack.isEmpty();
}
}