来自北大算法课的Leetcode题解：155. 最小栈

doubleZ0108

已于 2022-12-31 14:58:21 修改

阅读量90

点赞数

分类专栏：来自北大算法课的Leetcode题解文章标签： leetcode 算法

于 2022-08-20 15:34:48 首次发布

本文链接：https://blog.csdn.net/double_ZZZ/article/details/126440499

版权

来自北大算法课的Leetcode题解专栏收录该内容

247 篇文章 5 订阅

订阅专栏

本文介绍了一个Python类`MinStack`，该类扩展了标准栈的功能，允许在常数时间内获取栈内的最小元素。在`push()`、`pop()`和`getMin()`方法中维护最小值，确保高效操作。`push()`时比较并更新最小值，`pop()`时检查并更新最小值，而`getMin()`则直接返回最小值。

摘要由CSDN通过智能技术生成

代码仓库：Github | Leetcode solutions @doubleZ0108 from Peking University.

解法1(T92% S90%)：数据结构设计题。在栈的基础上能在常数时间获取到最小值，因此肯定需要把最小值存下来，在push和pop时维护，这里使用一个类内变量存储
- push()：加到栈顶，如果比当前最小值小则更新
- pop()：弹出栈顶，如果此时弹出的就是最小值则一次遍历更新最小值，如果栈为空则将最小值归为None
- top()：直接返回栈顶（题目保证在栈非空时调用）
- getMin()：直接返回最小值（题目保证在栈非空时调用）

class MinStack(object):

    def __init__(self):
        self.stack = []
        self.minVal = None


    def push(self, val: int) -> None:
        self.stack.append(val)
        if self.minVal is None or val < self.minVal:
            self.minVal = val


    def pop(self) -> None:
        re = self.stack.pop(-1)
        if re == self.minVal:
            if self.stack != []:
                self.minVal = self.stack[-1]
                for val in self.stack:
                    if val < self.minVal: self.minVal = val
            else: self.minVal = None
        


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


    def getMin(self) -> int:
        return self.