leetcode热题100.最小栈

作者：晓宜
🌈🌈🌈
个人简介：互联网大厂Java准入职，阿里云专家博主，csdn后端优质创作者，算法爱好者
🌙🌙🌙
今天给大家讲解下最小栈这一题，希望对你有所帮助
❤️❤️❤️
你的关注是我前进的动力😊

Problem: 155. 最小栈

题目

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

实现 MinStack 类:

• MinStack() 初始化堆栈对象。
• void push(int val) 将元素val推入堆栈。
• void pop() 删除堆栈顶部的元素。
• int top() 获取堆栈顶部的元素。
• int getMin() 获取堆栈中的最小元素。

示例 1:

输入： [“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.

解题方法

维护一个最小栈，记录当前进栈数字对应的栈中的最小值得，解法的要点是每一次进栈都要比较最小栈的栈顶元素与当前元素，这二者的最小值进栈
注意：最小栈一开始为一个极大值，这样第一次进栈的时候才能把第一个元素压进去

复杂度

时间复杂度:

$O(n)$

空间复杂度:

只申明了两个栈，所以用到了 $2n$的空间 $O(n)$

Code

class MinStack:

    def __init__(self):
        self.stack = []
        self.min_stack = [math.inf]


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


    def pop(self) -> None:
        self.min_stack.pop()
        return self.stack.pop()

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


    def getMin(self) -> int:
        return self.min_stack[-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()