用stack实现min stack

遇到个好玩的问题，就是用一个stack实现min stack，什么意思呢，就是我实现stack，但是能以O(1)的时间复杂度和空间复杂度去找到我stack里面的最小值。

常规的方法是：用一个变量存放当前最小值，但是会出现这种情况，就是当我的stack pop掉的值刚好是最小值时候，后面就没法知道当前的最小值了。

怎么办呢？可以考虑在push阶段做个改变，就是在我每次往stack里面push数据的时候，跟当前的最小值比较，如果比当前最小值还小的话，那么将当前最小值入栈，再把最小值修改为这个值。在pop阶段，如果遇到pop出来的值刚好等于当前的最小值，那么将下一个值pop出来，并将pop出来的值设置为当前栈中最小值，这个值就是在push阶段时放进去的，表明这个位置的下面位置都比这个值小。

 

附录python源码：

 

class MinStack(object):

    def __init__(self):

        self.stack = []
        self.min = 0

    def push(self, m):

        if(len(self.stack) == 0):
            self.min_ = m
            self.stack.append(m)
        else:
            if m <= self.min_:
                self.stack.append(self.min_)
                self.min_ = m
                self.stack.append(m)
            else:
                self.stack.append(m)

    def pop(self):

        if self.stack[len(self.stack) - 1] == self.min_:
            self.stack.pop()
            self.min_ = self.stack.pop()
        else:
            self.stack.pop()

    def get_min(self):
        return self.min_


if __name__ == '__main__':
    min_stack = MinStack()
    min_stack.push(4)
    min_stack.push(5)
    min_stack.push(7)
    min_stack.push(6)
    min_stack.push(-3)
    min_stack.push(-6)
    min_stack.push(18)
    min_stack.pop()
    min_stack.pop()
    print min_stack.get_min()