问题参考Leetcode 155,该题的关键在于设计一个能在常数时间检索到最小元素的栈,因此,我们需要维护额外一个一直记录最小元素的栈,这样才能常数时间检索。
class MinStack:
def __init__(self):
self.stack = []
self.minstack = []
def push(self, val: int) -> None:
self.stack.append(val)
self.minstack.append(min(self.minstack[-1], val)) if self.minstack else self.minstack.append(val)
def pop(self) -> None:
self.stack.pop()
self.minstack.pop()
def top(self) -> int:
return self.stack[-1]
def getMin(self) -> int:
return self.minstack[-1]