包含min函数的栈-python

原创于 2019-12-20 16:44:43 发布 · 166 阅读

CC 4.0 BY-SA版权

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

#思路：新建一个辅助栈来存储当前最小值，当有更小值入栈时，将其压入此栈，最小值为栈顶元素。当栈顶的元素出栈后，此时栈顶的元素就顺应成为新的最小值
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]