21包含min函数的栈python

题目：定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。
思路：设置一个辅助栈来保存最小值，每次在原始栈压入元素的时候同时更新辅助栈

class Solution:
    def __init__(self):
        self.stack = []
        self.min_stack = []
        
    def push(self, node):
        # write code here
        self.stack.append(node) #如果新压入的值比最小值小，则向辅助栈压入新值，否则压入最小值
        if self.min_stack == [] or node < self.min():
            self.min_stack.append(node)
        else:
            temp = self.min()
            self.min_stack.append(temp)
            
    def pop(self):
        # 只要不为空就删除原始栈和辅助栈的元素
        if self.min_stack == [] or self.stack == []:
            return None
        self.min_stack.pop()
        self.stack.pop()
        
    def top(self):#返回栈顶
        # write code here
        return self.stack[-1]
        
    def min(self):
        # 返回当前最小值
        return self.min_stack[-1]