20.包含min的函数

1.题目

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（在该栈中，调用min、push及pop的时间复杂度应为O(1)）。

2.思路

把每次压栈时的最小元素（每次新压栈元素和之前的最小元素进行比较）放进另外一个专门存放最小值的辅助栈中。例如：
（1）初始空栈，压入元素3，目前3是最小值，放入辅助栈，目前数据栈3，辅助栈3；
（2）压入元素4，因为4>3，所以仍然往辅助栈中压入3，目前数据栈3， 4，辅助栈中为3，3；
（3）压入元素2，因为2<3，所以往辅助栈中压入2， 目前数据栈3，4，2，辅助栈中为3，3，2；
（4）压入元素1， 因为1<2，所以往辅助栈中压入1，目前数据栈3，4，2，1，辅助栈中为3，3，2，1；
（5）弹出1，数据栈3，4，2，辅助栈中为3，3，2；
（6）弹出2，数据栈3，4，辅助栈中为3，3；
（7）压入元素0，数据栈3，4，0，辅助栈中为3，3，0.

3.代码

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.minStack = []
    def push(self, node):
        # write code here
        self.stack.append(node)
        if not self.minStack or node < self.minStack[-1]:
            self.minStack.append(node)  # 保存最小元素到栈顶
        else:
            self.minStack.append(self.minStack[-1])
    def pop(self):
        # write code here
        if self.stack:
            self.stack.pop()
            self.minStack.pop()
    def top(self):
        # write code here
        if self.stack:
            return self.stack[-1]
        else:
            return None
    def min(self):
        # write code here
        if self.minStack:
            return self.minStack[-1]
        else:
            return None