刷题记录 leetcode155：最小值栈

Posted 2020-12-18 嫩西瓜

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

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

 

思路：用一个辅助栈的栈顶记录原始栈的最小值，并且两个栈的高度保持相同，入栈时，当前值与辅助栈的栈顶比较，将两者的最小值入栈，这样能保证，在原始栈的最小值弹出前，辅助栈一直保存着最小值。

class MinStack {

    private Stack<Integer> dataStack;
    private Stack<Integer> minStack;
    private int min;

    public MinStack() {
        dataStack = new Stack<>();
        minStack = new Stack<>();
        min = Integer.MAX_VALUE;
    }

    public void push(int x) {
        dataStack.add(x);
        min = Math.min(min, x);
        minStack.add(min);
    }

    public void pop() {
        dataStack.pop();
        minStack.pop();
        min = minStack.isEmpty() ? Integer.MAX_VALUE : minStack.peek();    // 这句话是为了防止两种情况：1. minStack弹出后栈为空，
min必须更新为最大整数，否则min会保持原来栈的最小值，产生错误。
2. 如果弹出后栈不为空，但是弹出的恰好是当前的最小值，
如果min不更新为栈顶元素（新栈的最小值），也会产生错误
    }

    public int top() {
        return dataStack.peek();
    }

    public int getMin() {
        return minStack.peek();
    }
}