leetcode946. 验证栈序列 && 剑指 Offer 31. 栈的压入、弹出序列

题目链接

题目描述

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

示例 1：

输入：pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出：true
解释：我们可以按以下顺序执行：
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1

示例 2：

输入：pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
输出：false
解释：1 不能在 2 之前弹出。

提示：

• 0 <= pushed.length == popped.length <= 1000
• 0 <= pushed[i], popped[i] < 1000
• pushed 是 popped 的排列。

解题思路

使用一个栈stack,变量j初值为0表示popped数组的当前位置，我们遍历数组pushed，

• 当数组当前值pushed[i]不等于popped[j],就将pushed[i]压入栈stack中
• 相等则j++,然后循环当stack.size()>0&&stack.peek()=popped[j]的时候
  • 如果栈顶值等于popped[j]，就j++,弹出栈中元素

最后判断stack的大小是否为0，为0返回true,否则返回false

代码

class Solution {
    
    
    public boolean validateStackSequences(int[] pushed, int[] popped) {
    
    
        int m = pushed.length;

        Deque<Integer> deque = new LinkedList<>();
        int j = 0;
        for (int i = 0; i < m; i++) {
    
    
            if (popped[j] == pushed[i]) {
    
    
                j++;
                while (deque.size() > 0 && deque.peek() == popped[j]) {
    
    
                    deque.pop();
                    j++;
                }
            } else {
    
    
                deque.push(pushed[i]);
            }
        }
        return deque.size() > 0 ? false : true;
    }
}

时间复杂度

O(n)