Leetcode # 155：最小栈[easy]

描述

设计一个支持 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.

思路

因为要在O(1)时间复杂度完成查找最小元素，因此需要另一个容器来存储当前最小元素，并随着栈的入栈出栈动态更新。

• 可以利用两个栈来实现，一个栈用来存储数据，一个栈用来存储最小元素
• 入栈时，如果最小元素栈为空，则当前元素最小，入栈，如果不为空，则判断当前元素是否小于等于栈顶元素，如果是则入栈
• 出栈时，判断最小栈的栈顶元素是否是当前出栈的元素，如果是，则出栈。否则不作处理
• 之所以添加等于的条件，是因为可能碰到连续多个值相等，是最小值，之后有一个出栈的情况

代码

Java

class MinStack {
    
    //存储数据
    private Stack<Integer> dataStack;
    
    //存储最小值
    private Stack<Integer> minStack;

    /** initialize your data structure here. */
    public MinStack() {
        dataStack = new Stack<Integer>();
        minStack = new Stack<Integer>();
    }
    
    public void push(int x) {
        dataStack.push(x);
        if(minStack.isEmpty() || minStack.peek()>=x)
            minStack.push(x);
    }
    
    public void pop() {
        int top = dataStack.pop();
        if(minStack.peek() == top){
            minStack.pop();
        }
    }
    
    public int top() {
        return dataStack.peek();
    }
    
    public int getMin() {
        return minStack.peek();
    }
}

Python

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.data = []
        self.min = []

    def push(self, x: int) -> None:
        self.data.append(x)
        if len(self.min) == 0 or x <= self.min[-1]:
            self.min.append(x)

    def pop(self) -> None:
        top = self.data.pop()
        if top == self.min[-1]:
            self.min.pop()

    def top(self) -> int:
        return self.data[-1]
        

    def getMin(self) -> int:
        return self.min[-1]