剑指 Offer 30. 包含min函数的栈--python3

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.

代码：

class MinStack:

    def __init__(self):
        self.A=[]
        self.mins = [2147483647]
    def push(self, x: int) -> None:
        self.A.append(x)
        if x < self.mins[-1]:
            self.mins.append(x)
        else:
            self.mins.append(self.mins[-1])
    def pop(self) -> None:
        if self.A:
            self.A=self.A[:-1]
            self.mins=self.mins[:-1]
    def top(self) -> int:
        return self.A[-1]
    def min(self) -> int:
        return self.mins[-1]


minStack=MinStack()
minStack.push(2)
minStack.push(0)
minStack.push(3)
minStack.push(0)
print(minStack.min())
minStack.pop()
print(minStack.min())
minStack.pop()
print(minStack.min())
minStack.pop()
print(minStack.min())

输出：
0
0
0
2

解：
本题的方法是创建一个mins数组便于min函数的返回，mins中倒数第一个数用于存放A数组的最小值。方法是每push一个数 A和mins数组都必须有增加，A数组直接加，mins需要判断与mins数组前一个数的大小，如果比前一个数大，则重复添加mins数组的倒数第一个数，如果比其小，则将此数添加进去，直到出现比他还小的数就将它加入。以此类推。pop时，A数组始终与mins数组同进退，即A数组出栈时，mins也要相应出栈，A数组始终包含于mins数组。注意：由于mins数组刚开始没有元素，即第一次push时没有元素用于比较大小。所以将int类型的最大值加入其中。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof