栈的压入弹出序列(Java)
题目:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
思路:
把pushed序列模拟入栈,当遇到栈顶元素等于poped[idx],模拟出栈过程,如果能匹配(idx能指向序列末尾),那么poped是压栈序列的弹出序列。
代码:
class Solution {
public boolean validateStackSequences(int[] pushed, int[] popped) {
if(pushed == null || popped == null || pushed.length != popped.length){
return false;
}
Stack<Integer> stack = new Stack<>();
int num = popped.length;
int idx = 0;
for(int e : pushed){
stack.push(e);
while(idx < num && !stack.isEmpty() && stack.peek() == popped[idx]){
stack.pop();
idx++;
}
if(idx >= num) break;
}
return idx == num;
}
}