题目链接
https://leetcode-cn.com/problems/min-stack/
题目描述
代码初步
- 思路:
- 借用一个辅助栈min_stack,用于存储stack中最小值:
- push:每当push新值进来时,如果“小于等于”min_stack栈顶值,则一起push到min_stack,即更新了最小值;
- pop:判断pop出去的元素值是否是min_stack栈顶元素值(即最小值),如果是则将min_stack栈顶元素一起pop,这样可以保证min_stack栈顶元素始终是stack中的最小值。
getMin:返回min_stack栈顶即可。 - min_stack的作用是对stack中的元素做标记,标记的原则是min_stack中元素一定是降序的(栈底最大栈顶最小)。换个角度理解,min_stack等价于遍历stack所有元素,把升序的数字都删除掉,留下一个从栈底到栈顶降序的栈。本题要求获取最小值的复杂度是O(1),因此须构建辅助栈,在push与pop的过程中始终保持辅助栈为一个降序栈。
时间空间复杂度都为O(N),获取最小值复杂度为O(1)
**问题:**注意push时,是x<=self.minstack[-1],刚开始我是直接<,提交后发现测试不通过。因为当minstack列表只有一个值时,再执行pop()后,如果pop的为最小值,minstack是需要同步pop的,那么此时minstack就为空了,再执行getMin()就会报错。
例: [“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“getMin”]
[[],[0],[1],[0],[],[],[]]
class MinStack(object):
def __init__(self):
"""
initialize your data structure here.
"""
self.stack=[]
self.minstack=[]
def push(self, x):
"""
:type x: int
:rtype: None
"""
self.stack.append(x)
if not self.minstack or x<=self.minstack[-1]:
self.minstack.append(x)
def pop(self):
"""
:rtype: None
"""
if self.stack.pop()==self.minstack[-1]:
self.minstack.pop()
def top(self):
"""
:rtype: int
"""
return self.stack[-1]
def getMin(self):
"""
:rtype: int
"""
return self.minstack[-1]