提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  其它

当前栏目

最小栈（155）题解

最小 题解 155
2023-06-13 09:14:29 时间

题目

分析

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

push操作：

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

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

代码

辅助栈初始化时需要写入一个最大值，否则第一次比对栈顶没有内容会报错。
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();
    }
}

相关文章