用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()

  

原文地址:https://www.cnblogs.com/zihaowang/p/5781879.html