题目
设计一个支持 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.
提示:
-2^31 <= val <= 2 ^31 - 1
pop、top 和 getMin 操作总是在 非空栈 上调用
push, pop, top, and getMin最多被调用 3 * 10^4 次
来源:力扣(LeetCode)
解题思路
这个题肯定是不能在调用getMin的时候才计算最小值,否则将会超时。一个常用的思路就是将当前栈的最小值和栈顶的元素绑在一起,这样的话对于栈不管怎么操作栈顶元素都会维护一个当前栈的最小值。
class MinStack:def __init__(self):self.stack=[]def push(self, val: int) -> None:if self.stack:if val>=self.getMin():self.stack.append((val,self.getMin()))else:self.stack.append((val,val))else:self.stack.append((val,val))def pop(self) -> None:return self.stack.pop()[0]def top(self) -> int:return self.stack[-1][0]def getMin(self) -> int:return self.stack[-1][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()