最小栈
1,程序简介
设计一个支持 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 操作总是在 非空栈 上调用。
2,程序代码
"""
日期: Tue Mar 8 18:37:27 2022
功能: 最小栈
作者: 小梁aixj
"""
class MinStack:
def __init__(self):
self.data = [(None, float("inf"))]
def push(self, x: "int") -> "None":
self.data.append((x, min(x, self.data[-1][1])))
def pop(self) -> "None":
if len(self.data) > 1:
self.data.pop()
def top(self) -> "int":
return self.data[-1][0]
def getMin(self) -> "int":
return self.data[-1][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.getMin()