问题描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函 数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
实例
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/bao-han-minhan-shu-de-zhan-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思考
刚开始做的时候觉得很简单,不就是保存最小值吗,弄个变量保存就完了。结果人家要求所有的时间复杂度都是O(1)
如果我用单独的变量保存最小值的话,如果最小值被pop掉了该怎么办,找第二个小的?,这就需要同时保存第二小的值,但如果第二小的值比第一小的值先pop掉怎么办?瞬间陷入死循环
到这我肯定知道需要辅助的数据结构来保存一串变量,但怎么保存是一个问题,首先不能排序,因为时间复杂度已经超了
其次我看题目难度是简单,说明数据结构没那么复杂
解决这个问题的关键在于min是动态的,我们需要在pop和push操作时刻更新min的值,而单一变量的保存不可能将复杂度降至1·
解决
首先说明我还是看了答案,就是leecode解析里面的动画,奈何我太菜,还得需要提示才行
我纠结的点在于如何动态的获取并保存最小值,实际上最小值的获取与保存是静态的,他们在栈中的每一个状态都确定好了
栈可以抽象成一个可以溯源的状态序列,也就是说我们从状态1到状态2之后还可以回到状态1(这是由先进后出的性质决定的)
例如:[1] — [1,2] — [1]
此时我们就可以保存每个状态的参数,用于溯源做准备,这个用来保存参数的列表就用解析里面说的辅助栈
这里该状态的最小值与上一个状态的最小值有关,复杂度为O(1)
代码
class MinStack:
# 重点:状态序列的保存
def __init__(self):
"""
initialize your data structure here.
"""
self.a=[]
self.min_list=[]
def push(self, x: int) -> None:
self.a.append(x)
if self.min_list==[]:
self.min_list.append(x)
elif self.min_list[-1]>x:
self.min_list.append(x)
else:self.min_list.append(self.min_list[-1])
def pop(self) -> None:
if self.a!=[]:
self.a.pop()
self.min_list.pop()
def top(self) -> int:
return self.a[-1]
def min(self) -> int:
return self.min_list[-1]
延申
我们可以利用栈来实现状态序列的溯源,同时如果是目标是据结构是栈的话我们也应该考虑到这种性质,是否能够解决相应问题