最小栈（155）题解

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第6天，点击查看活动详情

题目

分析

此题中的push、pop、top操作都是Stack及Deque等类可以支持的，它们的时间复杂度为O(1)。难点在于getMain方法，获取堆栈中的最小元素，如果遍历整个堆栈区找寻最小值时间复杂度为O(n)，想要将时间复杂度优化到O(1)就需要辅助栈。

push操作：

辅助栈跟原始栈的操作是一致的，但是辅助栈的数据需要取push值跟当前辅助栈值的最小值，这样在获取栈内最小值时直接从辅助栈进行peek获取栈顶数据即可。

辅助栈的步骤跟原始栈的步骤一致，push跟pop都同步入栈跟出栈，要达到两个栈操作的一致。pop操作就简单了，不需要判断直接两个栈进行pop即可，就不画图了，看下面代码。

实现

辅助栈初始化时需要写入一个最大值，否则第一次比对栈顶没有内容会报错。

• push()：加到栈顶，如果比当前最小值小则更新
• pop()：弹出栈顶，如果此时弹出的就是最小值则一次遍历更新最小值，如果栈为空则将最小值归为None
• top()：直接返回栈顶（题目保证在栈非空时调用）
• getMin()：直接返回最小值（题目保证在栈非空时调用）

代码

class MinStack {

    ArrayDeque<Integer> stack;
    ArrayDeque<Integer> minStack;

    public MinStack() {
        stack = new ArrayDeque<>();
        minStack = new ArrayDeque<>();
        minStack.push(Integer.MAX_VALUE);
    }

    public void push(int val) {
        stack.push(val);
        minStack.push(Math.min(minStack.peek(), val));
    }

    public void pop() {
        stack.pop();
        minStack.pop();
    }

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

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