[LintCode 12] 带最小值操作的栈(Python)

最新推荐文章于 2025-05-20 17:26:39 发布

yurenguowang

最新推荐文章于 2025-05-20 17:26:39 发布

阅读量723

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： python 栈

本文链接：https://blog.youkuaiyun.com/yurenguowang/article/details/77840063

算法专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种特殊栈的实现方法，该栈除了支持常规的push和pop操作外，还支持O(1)时间复杂度的min操作，用于获取栈内最小值。通过维护两个栈，一个用于存储所有元素，另一个作为辅助栈来跟踪当前最小值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

实现一个带有取最小值min方法的栈，min方法将返回当前栈中的最小值。
你实现的栈将支持push，pop 和 min 操作，所有操作要求都在O(1)时间内完成。

注意事项
如果堆栈中没有数字则不能进行min方法的调用

样例
如下操作：push(1)，pop()，push(2)，push(3)，min()， push(1)，min() 返回 1，2，1

思路

push，pop操作都很简单，用一个列表模拟栈即可。min操作我们可以使用一个辅助栈，push和pop操作与栈同步，不同的是这个辅助栈存储的是当前最小的元素。

代码

class MinStack(object):
    def __init__(self):
        # do some intialize if necessary
        self.s = []
        self.m = []

    def push(self, number):
        # write yout code here
        self.s.append(number)
        if len(self.m) == 0:
            self.m.append(number)
        else:
            self.m.append(min(number, self.m[-1]))

    def pop(self):
        # pop and return the top item in stack
        self.m.pop()
        return self.s.pop()

    def min(self):
        # return the minimum number in stack
        return self.m[-1]