python每日一题【剑指 Offer 30. 包含 min 函数的栈】

day5-2022.10.29

昨天实际做了一题，但没有发存货
题目信息来源
作者：Krahets
链接：https://leetcode.cn/leetbook/read/illustration-of-algorithm
来源：力扣（LeetCode）

剑指 Offer 30. 包含 min 函数的栈

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.

题解：辅助栈法

这个题的难点可能是首先题目里显示了时间复杂度，故而不能遍历求min，考虑用来保存一个当前min，但是如果stack里面的min被弹出了，min又没有了，所以还需要保存历史min。

和另外一个用栈实现队列的很像。这里就是用一个min_stack用来存储当前来说的最小节点，即min_stack栈顶元素为对应stack元素的最小值。直到min被弹出，下一轮的min_stack栈顶元素为对应stack元素最小值

官方解析里的非严格排序是为了应对数重复的情况。

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.stack = []
        # 设置min栈用来存储最小值，如果最小值被出栈了，还有备份
        self.min_stack = []


    def push(self, x: int) -> None:
        self.stack.append(x)
        # 如果min栈为空或则min的栈顶元素大于新的元素，则min也append
        if not self.min_stack or self.min_stack[-1]>=x:
            self.min_stack.append(x)
        # 废了，min_top元素和stack top元素不一致，stack top被pop了，但是min里面的该值还被压在下面
        # else:
        #    
        #     min_num = self.min_stack.pop()
        #     # 获得
        #     if (self.min_stack and x<=self.min_stack[-1]) or not self.min_stack:
        #         self.min_stack.append(x)
        #     self.min_stack.append(min_num)

    def pop(self) -> None:
        pop_num = self.stack.pop()
        if self.min_stack[-1]==pop_num:
            self.min_stack.pop()
        

    def top(self) -> int:
        return self.stack[-1] if self.stack else None

    def min(self) -> int:
        return self.min_stack[-1] if self.min_stack else None

# 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()