最小栈(155)题解

题目

在这里插入图片描述

分析

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

push操作

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

辅助栈的步骤跟原始栈的步骤一致,pushpop都同步入栈出栈,要达到两个栈操作的一致。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();
    }
}

猜你喜欢

转载自blog.csdn.net/AnNanDu/article/details/126847127
今日推荐