题目:
设计一个支持push,pop,top,操作,并能在常数时间内检索到最小元素的栈。
辅助栈法:
1:使用两个栈,一个主栈用于存储所有元素,另一个辅助栈用于存储当前元素的最小值
2: 每次push时,将元素压入主栈,同时比较当前元素与辅助栈顶元素,将较小者压入辅助栈
3:每次pop时,同时从主栈和辅助栈弹出元素
4:getmin,直接返回辅助栈最小的栈顶元素
class MinStack(object):
def __init__(self):
self.stack=[]
self.min_stack=[]
def push(self, val):
"""
:type val: int
:rtype: None
"""
self.stack.append(val)
if not self.min_stack or val <=self.min_stack[-1]:
self.min_stack.append(val)
def pop(self):
"""
:rtype: None
"""
if self.stack:
top_val=self.stack.pop()
if top_val==self.min_stack[-1]:
self.min_stack.pop()
def top(self):
"""
:rtype: int
"""
if self.stack:
return self.stack[-1]
def getMin(self):
"""
:rtype: int
"""
if self.min_stack:
return self.min_stack[-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()