python-leetcode 69.最小栈
题目:
设计一个支持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()