[LeetCode] 155. Min Stack 最小栈

╰+攻爆jí腚メ 2022-10-02 12:52 130阅读 0赞

设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。

push(x) -- 将元素 x 推入栈中。  
    pop() -- 删除栈顶的元素。  
    top() -- 获取栈顶元素。  
    getMin() -- 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();  
minStack.push(-2);  
minStack.push(0);  
minStack.push(-3);  
minStack.getMin();   --> 返回 -3.  
minStack.pop();  
minStack.top();      --> 返回 0.  
minStack.getMin();   --> 返回 -2.

来源：力扣（LeetCode）  
链接：https://leetcode-cn.com/problems/min-stack  
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法：只使用1个栈，用一个变量min\_val记录当前的最小值，将当前最小值一同入栈，为节省空间，仅在当前最小值更改时才入栈。所以该栈的push和pop实际上可能是两次。当新进来的数小于min\_val时，把当前的min\_val和新进来的数一起推入到栈，min\_val变为这个新进来的数。当pop栈顶元素的时候，如果栈顶元素的值和min\_val相等，那么就把它下面记录的之前最小值赋给min\_val并弹出。

python代码：

class MinStack(object):
     
        def __init__(self):
            self.min = 2147483647
            self.stack = []
     
        def push(self, x):
            if x <= self.min:
                self.stack.append(self.min)
                self.min = x
            self.stack.append(x)
     
        def pop(self):
            peak = self.stack.pop()
            if peak == self.min:
                self.min = self.stack.pop()
     
        def top(self):
            return self.stack[-1]
     
        def getMin(self):
            return self.min