设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
来源:力扣(LeetCode)155题
解法一
双栈,记录每一步的最小值,pop维护最小值栈
class MinStack:
def __init__(self):
"""
initialize your data structure here.
"""
self.minv = []
self.s = []
def push(self, x: int) -> None:
self.s.append(x)
if not self.minv:
self.minv.append(x)
elif x <= self.minv[-1]:
self.minv.append(x)
def pop(self) -> None:
if self.s:
if self.s[-1] == self.minv[-1]:
self.minv.pop()
self.s.pop()
else:
return False
def top(self) -> int:
if self.s:
return self.s[-1]
else:
return False
def getMin(self) -> int:
if self.minv:
return self.minv[-1]
else:
return False
# 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()
解法二
双向链表【待实现】