力扣(leetcode) 155. 最小栈 (辅助栈法)

题目在这: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]

你可能感兴趣的:(个人笔记,交流学习,leetcode,python)