题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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.
提示:
各函数的调用总次数不超过 20000 次
思路
题目难在要求调用min函数的时间复杂度为O(1)。直接调用库中min函数的时间复杂度是O(n)。
需要两个栈,栈1用来入栈出栈操作,栈2用来非全部元素的排序,栈底到栈顶从大到小排序,要求输出最小值时就是栈顶的值。
为什么说是非全部元素呢,第一个元素压入栈1,同时压入栈2,当栈1为空的时候栈2才会为空。后来压入栈1的元素小于等于栈2的栈顶时才能压入栈2,这就保持了栈2中的元素按数值大小排序,栈顶是最小的元素。
小于等于中等于的原因是:保持两栈元素的一致性,调用pop函数出栈1时,如果此时是栈2的栈顶元素,那么栈2也pop。如果没有等于,那么当栈1压入多个相同的最小值,此时栈2只有1个最小值,那么就无法使用pop保持两栈的元素一致性了。
代码
class MinStack:
def __init__(self):
"""
initialize your data structure here.
"""
self.stack = []
self.stack2 = [] #辅助栈,帮助非完全元素从栈底到栈顶从大到小排序
def push(self, x: int) -> None:
self.stack.append(x)
if not self.stack2 or self.stack2[-1] >= x: #栈2为空或者栈2的栈顶比x大
self.stack2.append(x) #x成为新栈顶
def pop(self) -> None:
#self.stack.pop()
if self.stack.pop() == self.stack2[-1]: ##self.stack.pop会执行
self.stack2.pop() #保持两栈元素一致性
def top(self) -> int:
return self.stack[-1]
def min(self) -> int:
return self.stack2[-1]
# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.min()
知识点
def pop(self) -> None:
if self.stack.pop() == self.stack2[-1]:
self.stack2.pop()
此时self.stack.pop()会执行语句,就不用再单独一句了,否则会出错。