【力扣：python刷题】剑指 Offer 30. 包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.

提示：
各函数的调用总次数不超过 20000 次

思路

题目难在要求调用min函数的时间复杂度为O(1)。直接调用库中min函数的时间复杂度是O(n)。

需要两个栈，栈1用来入栈出栈操作，栈2用来非全部元素的排序，栈底到栈顶从大到小排序，要求输出最小值时就是栈顶的值。

为什么说是非全部元素呢，第一个元素压入栈1，同时压入栈2，当栈1为空的时候栈2才会为空。后来压入栈1的元素小于等于栈2的栈顶时才能压入栈2，这就保持了栈2中的元素按数值大小排序，栈顶是最小的元素。

小于等于中等于的原因是：保持两栈元素的一致性，调用pop函数出栈1时，如果此时是栈2的栈顶元素，那么栈2也pop。如果没有等于，那么当栈1压入多个相同的最小值，此时栈2只有1个最小值，那么就无法使用pop保持两栈的元素一致性了。

代码

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.stack = []
        self.stack2 = [] #辅助栈，帮助非完全元素从栈底到栈顶从大到小排序

    def push(self, x: int) -> None:
        self.stack.append(x)
        if not self.stack2 or self.stack2[-1] >= x: #栈2为空或者栈2的栈顶比x大
            self.stack2.append(x)  #x成为新栈顶

    def pop(self) -> None:
        #self.stack.pop()
        if self.stack.pop() == self.stack2[-1]:  ##self.stack.pop会执行
            self.stack2.pop()  #保持两栈元素一致性

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

    def min(self) -> int:
        return self.stack2[-1]



# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.min()

知识点

def pop(self) -> None:
        if self.stack.pop() == self.stack2[-1]:  
            self.stack2.pop()  

此时self.stack.pop()会执行语句，就不用再单独一句了，否则会出错。