包含min函数的栈-python

看雨c

于 2019-12-20 16:44:43 发布

阅读量153

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_45947707/article/details/103635020

版权

本文介绍了一种实现带有获取最小元素操作的栈的方法。通过使用辅助栈存储当前最小值，可在常数时间内完成最小值查询。当新元素小于等于当前最小值时，将其压入辅助栈；否则，压入当前最小值。出栈时同步更新辅助栈。

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

#思路：新建一个辅助栈来存储当前最小值，当有更小值入栈时，将其压入此栈，最小值为栈顶元素。当栈顶的元素出栈后，此时栈顶的元素就顺应成为新的最小值
class min_stack:
    def __init__(self):
        self.stack = []
        self.minstack = []

    def push(self, data):
        # 将新元素添加到栈
        self.stack.append(data)
        # 当新元素比之前的元素小时，将新元素压入到辅助栈，否则将之前元素重复压入辅助栈
        if data < self.minstack[-1] or len(self.minstack) == 0:
            self.minstack.append(data)
        else:
        	self.minstack.append(self.minstack[-1])

    def pop(self):
        self.minstack.pop()
        return self.stack.pop()
    
    def min(self):
        return self.minstack[-1]