【剑指 Offer】—— 31. 栈的压入、弹出序列

题目

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如，序列 {1,2,3,4,5} 是某栈的压栈序列，序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列，但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。

题目来源：力扣（LeetCode）

思路

1.使用一个辅助栈stack模拟元素入栈出栈的过程
2.遍历pushed模拟入栈过程,如果在push过程中如果发现stack.peek()==popped[i]说明此时popped[i]元素应该出栈了
3.如果是弹出序列的话,到最后所有元素都push完之后正好所有元素也应该pop完,栈为空

举例

对于入栈序列 {1,2,3,4,5}和出栈序列 {4,3,5,1,2},在

            stack.push(num);
            while (!stack.isEmpty() && stack.peek() == popped[i]) {
                stack.pop();
                i++;
            }

这个过程中push到4的时候发现栈顶为4而popped[0]为4那么此时应该将4弹出

代码

 public boolean validateStackSequences(int[] pushed, int[] popped) {
        Stack<Integer> stack = new Stack<>();
        int i = 0;
        for (int num : pushed) {
            stack.push(num);
            while (!stack.isEmpty() && stack.peek() == popped[i]) {
                stack.pop();
                i++;
            }
        }
        return stack.isEmpty();
    }