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

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

push – 在顶部加入
pop – 删除最上面的元素
min – 找出栈中最小元素

思路：
栈就像积木一样，只能在尾部/顶部进行插入和删除操作。
因为要求时间复杂度为O(1)，min函数里不能用循环，只能采用辅助栈B。
A栈中正常添加元素，如果B栈中无元素或添加的那个元素小于等于B栈最上面那个元素，则把元素添加到B中，这样就能保持B栈最上面的元素始终是最小的元素。
这里一定是小于等于，因为pop()要删除A最后一个元素以及若A最后一个元素是B最上面的元素，则需删除B最上面的元素，所以若有两个相等的最小值，则只会删除一个。

python代码：

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.A = []
        self.B = []  # 辅助栈


    def push(self, x: int) -> None:
        self.A.append(x)
        if not self.B or self.B[-1] >= x:  # 这里一定是>=
            self.B.append(x)

    def pop(self) -> None:  # 若元素在B中同时也要删除B中元素
        if self.A.pop() == self.B[-1]:
            self.B.pop()

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

    def min(self) -> int:
        if self.B:
            return self.B[-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()