leetcoe--栈的压入、弹出序列

最新推荐文章于 2021-11-29 21:09:30 发布

原创最新推荐文章于 2021-11-29 21:09:30 发布 · 346 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #java #算法 #数据结构 #python

LeetCode 专栏收录该内容

146 篇文章

订阅专栏

本文探讨了如何判断给定的弹出序列是否为特定压栈序列的有效弹出顺序。通过模仿栈操作，使用Java和Python实现算法，确保所有数字不重复的情况下，检查序列的有效性。

题目来自LeetCode，链接：面试题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 之前弹出。

这道题目的做法其实就是模仿一个入栈出栈顺序，看是否能够得到给定的一个弹出序列，具体如下：

初始化i=0，j=0
将pushed[i]入栈
如果栈顶刚好等于popped[j]，则出栈，j=j+1，继续判断栈顶元素是否等于popped[j]，知道栈为空或者不再相等为止停止
当pushed的元素全部入栈完成后，只需看栈是否为空即可判断popped是否为对应的弹出序列了

JAVA版代码如下：

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        Stack<Integer> stack = new Stack<>();
        int popIdx = 0;
        for (int p : pushed) {
            if (p == popped[popIdx]) {
                ++popIdx;
                while (stack.size() > 0 && stack.peek() == popped[popIdx]) {
                    stack.pop();
                    ++popIdx;
                }
            }
            else {
                stack.push(p);
            }
        }
        return stack.size() == 0;
    }
}

提交结果如下:

Python版代码如下：

class Solution:
    def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
        popIdx = 0
        stack = []
        for p in pushed:
            if p == popped[popIdx]:
                popIdx += 1
                while len(stack) > 0 and stack[-1] == popped[popIdx]:
                    stack.pop()
                    popIdx += 1
            else:
                stack.append(p)
        return len(stack) == 0

提交结果如下: