LeetCode刷题笔记第155题:最小栈
想法:
需要实现栈中的入栈、出栈、获取顶部元素和栈中最小元素,使用辅助栈完成功能。
入栈操作直接在原基础上尾部添加
出栈操作删除栈顶的元素
获取顶部元素操作直接获取数组中的最后一个元素
栈中最小的元素会随栈中元素的不同动态变化,在元素入栈时就在辅助栈中记录当前栈的最小元素,随着元素的入栈出栈对辅助栈做相应的最小元素判断及入栈出栈变化,因此当前辅助栈中的栈顶元素(数组中的最后一个元素)即为当前栈的最小元素
class MinStack:
def __init__(self):
self.stack = []
self.min_stack = [math.inf]
def push(self, val: int) -> None:
self.stack.append(val)
self.min_stack.append(min(val, self.min_stack[-1]))
def pop(self) -> None:
self.stack.pop()
self.min_stack.pop()
def top(self) -> int:
return self.stack[-1]
def getMin(self) -> int:
return self.min_stack[-1]
# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(val)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.getMin()