1、题目描述:
2、方法:
思路:
思路来源:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/solution/mian-shi-ti-30-bao-han-minhan-shu-de-zhan-fu-zhu-z/
如果还有点懵逼,下面我说一下具体的做法就知道怎么回事了:
- 当元素入栈时,即执行 p u s h ( ) push() push()函数时,我们将元素压入数据栈 A A A中,同时检查当前辅助栈 B B B的情况:辅助栈 B B B为空或者当前要 p u s h ( ) push() push()的元素小于等于辅助栈 B B B栈顶的元素,则把当前元素也压入辅助栈 B B B。
- 当元素出栈时,我们把数据栈 A A A栈顶的元素弹出,同时检查数据栈 A A A栈顶元素是不是等于辅助栈 B B B栈顶元素,如果等于,则把辅助栈 B B B栈顶元素也出栈。
代码:
class MinStack:
def __init__(self):
"""
initialize your data structure here.
"""
self.stackA = [] # 数据栈
self.stackB = [] # 辅助栈
def push(self, x: int) -> None:
self.stackA.append(x)
if not self.stackB or self.stackB[-1] >= x:
self.stackB.append(x) # 如果stackB为空,或者x小于等于栈顶元素,则x直接入stackB
def pop(self) -> None:
# 栈B中存储栈A中所有非严格降序的元素,则栈A中的最小元素始终对应栈B的栈顶元素
if self.stackA.pop() == self.stackB[-1]:
self.stackB.pop()
def top(self) -> int:
return self.stackA[-1] # 栈顶元素
def min(self) -> int:
return self.stackB[-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()