LeetCode155. 最小栈

题目

最小栈
设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。

push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。

示例:

输入：
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]

输出：
[null,null,null,null,-3,null,0,-2]

解释：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

提示：

pop、top 和 getMin 操作总是在 非空栈 上调用。

代码

利用辅助栈来存储栈里面元素的最小值，辅助栈初始值为math.inf，方便第一次进行判断比较是否是最小值。每次入栈辅助栈入栈的值取当前要入栈的值和辅助栈栈顶的值两者的最小值。出栈操作都弹出栈顶元素。 top操作返回栈的栈顶元素，getMin操作返回辅助栈的栈顶元素

class MinStack:

    def __init__(self):
        self.stackA=[]
        self.min=[math.inf]

    def push(self, val: int) -> None:
        self.stackA.append(val)
        self.min.append(min(val,self.min[-1]))

    def pop(self) -> None:
        self.stackA.pop()
        self.min.pop()

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

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



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