剑指 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 的排列。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zhan-de-ya-ru-dan-chu-xu-lie-lcof

解题思路

注意点:压入栈的所有数字均不相等
本题思路就是使用一个辅助栈,来模拟入栈和出栈操作

• 创建一个辅助栈和初始化出栈数组下标index = 0
• 循环入栈数组,每次循环内,执行以下两步操作
  • 当前元素入栈
  • 循环判断当前栈不为空且栈顶元素和出栈数组下标index的值相等时
    • 执行出栈操作
    • 出栈数组下标index++
• 判断栈是否为空

代码

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        // 创建一个栈
        LinkedList<Integer> stack = new LinkedList<>();
        int index = 0;
        for (int item : pushed) {
            // 模拟入栈操作
            stack.push(item);
            // 当栈不为空且栈顶元素和出栈序列当前下标相等时
            while (!stack.isEmpty() && stack.peek() == popped[index]) {
                // 下标后移
                index++;
                // 出栈
                stack.pop();
            }
        }
        // 判断栈是否为空
        return stack.isEmpty();
    }
}

复杂度

• 时间复杂度:O(n),入栈和出栈最多2n次
• 空间复杂度:O(n),辅助栈最多存储n个元素