[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]