python 包含min函数的栈

剑指offer第30题：包含min函数的栈

题目：定义栈的数据结构，实现一个能够获得最小值的min函数。在该栈中，调用min、push及pop的时间复杂度都是O(1)。
这题如果用python的min函数，解决十分轻松，但可能不符合它的时间复杂度，我们自己实现栈，并完成min函数。

# _*_encoding:utf-8 _*_
class Stack:
    """
    用列表表示栈，符合先进后出的原则
    用两个栈解决：
    第一个栈stack正常出入栈
    第二个support_stack每次push时，将stack中的最小值push进去，而不是直接push某一个值
    每次pop操作，两个栈都进行pop,
    这样support_stack[-1]表示当前stack的最小值
    """

    def __init__(self):
        self.stack = []
        self.support_stack = []
        # min_num其实可以不要，用support_stack[-1]表示即可
        self.min_num = None

    def push(self, num):
        self.stack.append(num)
        if self.min_num is None:
            self.min_num = num
            self.support_stack.append(self.min_num)
        elif self.min_num < num:
                self.support_stack.append(self.min_num)
        else:
            self.min_num = num
            self.support_stack.append(num)

    def pop(self):
        if self.stack:
            self.support_stack.pop()
            return self.stack.pop()
        return 

    def min(self):
        return self.support_stack[-1]


if __name__ == '__main__':
    stack = Stack()
    current_stack = [11, 7, 8, 9, 5, 12, 15]
    for num in (current_stack):
        stack.push(num)
    print stack.min(), stack.stack
    # 测试用例
    stack.push(7)
    # 输入一个比小值大的
    print stack.min(), stack.stack
    stack.push(4)
    # 输入一个比小值小的 
    print stack.min(), stack.stack
    stack.pop()
    # 弹出最小值
    print stack.min(), stack.stack
    stack.pop()
    # 弹出非最小值
    print stack.min(), stack.stack