题目在这:https://leetcode-cn.com/problems/min-stack/
思路分析:
这道题乍一看挺吓人的,以往基本都是一个函数,这一下这么多函数,实际上每个函数里就一两行代码。
本题实现栈的功能,找到栈中最小值这么一个功能。
我们可以直接设置一个辅助栈,
1如果进原栈的数值比辅助栈顶小,则让他也进辅助栈。
2原栈出栈的时候,对比辅助栈顶,相等则吧辅助栈的元素也弹出去。
这样当想求最小值的时候。直接弹出辅助栈顶元素即可。
完整代码:
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]