[LintCode 12] 带最小值操作的栈(Python)

题目描述

实现一个带有取最小值min方法的栈，min方法将返回当前栈中的最小值。
你实现的栈将支持push，pop 和 min 操作，所有操作要求都在O(1)时间内完成。

注意事项
如果堆栈中没有数字则不能进行min方法的调用

样例
如下操作：push(1)，pop()，push(2)，push(3)，min()， push(1)，min() 返回 1，2，1

思路

push，pop操作都很简单，用一个列表模拟栈即可。min操作我们可以使用一个辅助栈，push和pop操作与栈同步，不同的是这个辅助栈存储的是当前最小的元素。

代码

class MinStack(object):
    def __init__(self):
        # do some intialize if necessary
        self.s = []
        self.m = []

    def push(self, number):
        # write yout code here
        self.s.append(number)
        if len(self.m) == 0:
            self.m.append(number)
        else:
            self.m.append(min(number, self.m[-1]))

    def pop(self):
        # pop and return the top item in stack
        self.m.pop()
        return self.s.pop()

    def min(self):
        # return the minimum number in stack
        return self.m[-1]