设计一个栈
设计一个支持push,pop,top,getMin操作,并且能在常数时间内检索最小元素的栈
push(x)
pop()
top()
getMin()
示例
输入:
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
关于解题
-
思路:利用以空间换时间的思想,创建一个辅助栈min_stack(),用于获取stack中的最小值。
-
算法流程
(1)创建两个自定义空栈,一个为原始栈stack(),一个为辅助栈min_stack()。
(2)关于push(),每当有新值入栈时,如果小于等于min_stack()栈顶值即最小 值时,则同时将新值push()进min_stack(),即同时更新栈顶最小值。
(3)关于pop(),需要判断stack.pop()是否是min_stack()的栈顶元素即最小值,如果是,则同时进行min_stack.pop(),则可以保证min_stack()的栈顶元素是stack()的最小值。
(4)关于top(),直接返回stack()栈顶元素即可。
(5)关于getMin(),直接返回min_stack()栈顶元素即可。 -
算法复杂度分析
时间复杂度O(1):入栈,出栈,获取值的所有操作,其时间复杂度都为O(1)。
空间复杂度O(n):n为总操作数,最坏情况为连续入栈n个数,此时两个栈占用的空间为O(n),即其包含n个元素辅助栈占用线性大小的额外空间。
code
class MinStack:
def __init__(self):
self.stack = []
self.min_stack = []
def push(self, x: int) -> None:
self.stack.append(x)
if not self.min_stack() or x <= self.min_stack[-1]: # 若辅助栈为空或新入栈元素比辅助栈栈顶值更小
self.min_stack.append(x)
def pop(self) -> None:
if self.stack.pop() == self.min_stack[-1]:
self.min_stack.pop()
def top(self) -> int:
return self.stack[-1]
def getMin(self) -> int:
return self.min_stack[-1]
code中部分问题参考
1.关于def __ init __(self):可参考__init __在python中的用法
2.push()代码中,若没有not self.min_stack( )的话,则忽略此时栈stack()为空的情况,即程序访问的self.min_stack[-1]是不存在的,则代码运行后会报越界错误,因此需要加判空条件。
3.pop()返回并删除栈顶元素。