设计一个堆栈,以支持在恒定时间内推送,弹出,顶部和检索最小元素。
- push(x)-将元素x推入堆栈
- pop()-删除堆栈顶部的元素。
- top()-获取顶部元素。
- getMin()-检索堆栈中的最小元素。
Input:["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]
Output:[null,null,null,null,-3,null,0,-2]
比较好理解的方法:建立一个空的list,然后push就用insert插入第一位,pop就返回除了第一个以外的值,top则返回list里面的第一个值。
'''
MinStack minStack = new MinStack()
minStack.push(-2); minStack.push(0)
minStack.push(-3); minStack.getMin() // return -3
minStack.pop(); minStack.top() // return 0
minStack.getMin() // return -2
'''
class MinStack:
def __init__(self):
"""
initialize your data structure here.
"""
self.stack=[]
def push(self, x: int) -> None:
self.stack.insert(0,x)
def pop(self) -> None:
self.stack=self.stack[1:]
def top(self) -> int:
return self.stack[0]
def getMin(self) -> int:
return min(self.stack)
# 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()
这个运行时间少一点
class MinStack:
def __init__(self):
self.val = []
self.min = float('inf') #初始值为正无穷
def push(self, x: int) -> None:
self.min = min(self.min, x) #每次都选取最小值
self.val.append([x, self.min]) #每次push进去x之后再加入当前最小值
def pop(self) -> None:
if len(self.val) > 1:
res = self.val.pop()
self.min = self.val[-1][1] #最后一位[-1]的第二个数[1]
return res
else:
self.min = float('inf')
return self.val.pop()
def top(self) -> int:
return self.val[-1][0]
def getMin(self) -> int:
return self.val[-1][1]