剑指offer 30：包含min函数的栈（Python）

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

class MinStack:

#push操作：第一个元素先全部放到栈A（入栈的元素）和B（存放小的元素）中，之后要如栈A的元素和栈B中top的袁术进行比较，小的存放到B中
#pop操作：A和B的顶端元素pop（只有把B中的顶端元素去掉，就是次小元素，下次输出）
#top操作：A的顶端元素
#min操作：B的顶端元素
    def __init__(self):
        self.stack_A=[]
        self.stack_B=[]

    def push(self, x: int) -> None:
        self.stack_A.append(x)
        if self.stack_B == []:
            self.stack_B.append(x)
        else:
            val = min(x,self.stack_B[-1])
            self.stack_B.append(val)

    def pop(self) -> None:
        self.stack_A.pop(-1)
        self.stack_B.pop(-1)


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


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