【剑指offer】19—包含min函数的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

链接：https://www.nowcoder.com/questionTerminal/4c776177d2c04c2494f2555c9fcc1e49?answerType=1&f=discussion
来源：牛客网

主流思路是应用于一个辅助栈，也就是最小元素栈。 每次压栈操作时, 如果压栈元素比当前最小元素更小, 就把这个元素压入最小元素栈, 原本的最小元素就成了次小元素. 同理, 弹栈时, 如果弹出的元素和最小元素栈的栈顶元素相等, 就把最小元素的栈顶弹出.

class Solution:
    def __init__(self):
        self.stack = []
        self.minvalue = []
    def push(self, node):
        # write code here
        self.stack.append(node) #栈保存节点值，把最小值存到列表中
        if self.minvalue:
            if self.minvalue[-1]>node:
                self.minvalue.append(node)
            else:
                self.minvalue.append(self.minvalue[-1])
        else:
            self.minvalue.append(node)

    def pop(self):
        # write code here
        if self.stack == []:
            return None
        self.minvalue.pop() #弹出的时候栈和最小值都有进行弹出
        return self.stack.pop()
    
        res.remove(res[-1])
    def top(self):
        # write code here
        if self.stack == []:
            return None
        return self.stack[-1]
    def min(self):
        # write code here
        if self.minvalue == []:
            return None
        return self.minvalue[-1]