遇到个好玩的问题,就是用一个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()
时间: 2024-12-26 15:17:05