栈的压入、弹出序列
题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
思路
- 需要一个辅助栈
- 如果下一个弹出栈的数字刚好是栈顶数字,那么直接弹出。若不是,把压栈序列中没有入栈的的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶为止。若所有数字都压入栈仍然没有找到下一个弹出的数字,那这个序列就不可能是弹出栈的序列
代码
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
boolean result = false;
if (pushA != null && popA != null && pushA.length != 0 && popA.length != 0) {
Stack<Integer> stack = new Stack<Integer>();
int pushNext = 0; // 压入序列的指针
int popNext = 0; // 弹出序列的指针
while (popNext < popA.length) {
while (stack.empty() || (stack.peek() != popA[popNext])) {
// 若栈顶不是要弹出的元素,就继续压栈
if (pushNext == pushA.length) {
break;
}
stack.push(pushA[pushNext]);
pushNext++;
}
if (stack.peek() != popA[popNext]) {
// 遍历完了都没找到弹栈元素,那就是没有
break;
}
// 找到之后栈顶弹出,继续找下一个
stack.pop();
popNext++;
}
if (stack.empty() && popNext == popA.length) {
// 该弹出的元素都弹出了,弹栈序列也遍历完了,那就是了
result = true;
}
}
return result;
}
}