【简单】155. 最小栈

原创已于 2022-02-25 16:21:31 修改 · 175 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #链表 #深度优先

于 2022-02-25 16:20:25 首次发布

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

这篇博客探讨了三种不同的Python实现MinStack的方法，对比了它们的执行时间和内存消耗。第一种方法使用两个栈，第二种方法利用元组存储最小值，第三种方法通过巧妙的差分计算节省空间。所有方法都成功通过了测试用例，而第三种方法在不占用额外辅助栈空间的情况下实现了高效性能。

题目
代码
执行用时：64 ms, 在所有 Python3 提交中击败了41.59% 的用户
内存消耗：18.2 MB, 在所有 Python3 提交中击败了43.94% 的用户
通过测试用例：31 / 31

class MinStack:
    def __init__(self):
        self.A,self.B=[],[]

    def push(self, val: int) -> None:
        self.A.append(val)
        if not self.B or val<=self.B[-1]:
            self.B.append(val)

    def pop(self) -> None:
        if self.A.pop()==self.B[-1]:
            self.B.pop()

    def top(self) -> int:
        return self.A[-1]

    def getMin(self) -> int:
        return self.B[-1]

【方法2】
执行用时：60 ms, 在所有 Python3 提交中击败了54.96% 的用户
内存消耗：18.9 MB, 在所有 Python3 提交中击败了5.10% 的用户
通过测试用例：31 / 31

class MinStack:
    def __init__(self):
        self.A=[]

    def push(self, val: int) -> None:
        if not self.A:
            self.A.append((val,val))
        else:
            self.A.append((val,min(val,self.A[-1][1])))

    def pop(self) -> None:
        self.A.pop()

    def top(self) -> int:
        return self.A[-1][0]

    def getMin(self) -> int:
        return self.A[-1][1]

【方法3】力推
不占用额外辅助栈的空间
执行用时：60 ms, 在所有 Python3 提交中击败了54.96% 的用户
内存消耗：18.4 MB, 在所有 Python3 提交中击败了26.99% 的用户
通过测试用例：31 / 31

class MinStack:
    def __init__(self):
        """
        initialize your data structure here.
        """
        self.stack = []
        self.min_value = -1

    def push(self, x: int) -> None:
        if not self.stack:
            self.stack.append(0)
            self.min_value = x
        else:
            diff = x-self.min_value
            self.stack.append(diff)
            self.min_value = self.min_value if diff > 0 else x

    def pop(self) -> None:
        if self.stack:
            diff = self.stack.pop()
            if diff < 0:
                top = self.min_value
                self.min_value = top - diff
            else:
                top = self.min_value + diff
            return top

    def top(self) -> int:
        return self.min_value if self.stack[-1] < 0 else self.stack[-1] + self.min_value

    def getMin(self) -> int:
        return self.min_value if self.stack else -1