剑指 Offer 30. 包含 min 函数的栈，时间复杂度O(1)（简单）

徐同学呀

已于 2022-04-28 13:43:06 修改

阅读量492

点赞数 3

分类专栏：刷他呀的算法文章标签：栈辅助栈

于 2022-04-28 13:01:22 首次发布

本文为博主原创文章，转载请注明本人源链接。微信公众号搜索：徐同学呀，愿与你共同进步！

本文链接：https://blog.csdn.net/weixin_36586120/article/details/124471220

版权

刷他呀的算法专栏收录该内容

4 篇文章 0 订阅

订阅专栏

该博客介绍了如何在Java中实现一个包含min函数的栈，使得push、pop、top和min操作的时间复杂度都是O(1)。通过维护两个栈，一个用于正常操作，另一个用于存储当前栈中的最小值，确保在主栈元素变化时，辅助栈始终能提供最小值。在push时，如果新元素小于或等于辅助栈顶元素，则将其压入辅助栈；在pop时，如果弹出的元素等于辅助栈顶元素，则辅助栈也弹出。这种设计保证了最小值栈的高效性。

摘要由CSDN通过智能技术生成

题干

剑指 Offer 30. 包含 min 函数的栈：
定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop、top 的时间复杂度都是 O(1)。

class MinStack {

    public MinStack() {

    }
    
    public void push(int x) {

    }
    
    public void pop() {

    }
    
    public int top() {

    }
    
    public int min() {

    }
}

原题链接：https://leetcode-cn.com/leetbook/read/illustration-of-algorithm/50bp33/

思路解析

普通栈push、pop、top的时间复杂度为O(1)，要想找到栈中的元素min，需要遍历整个栈进行比较，时间复杂度为O(n)，但是题干要求min操作也是O(1)。

就不能单纯使用一个栈实现了，可以利用另外一个栈存第一个栈中最小的元素。

需要保证第二个辅助栈的栈顶元素是第一个主栈中元素的min，如果min出栈了，辅助栈也要出栈，辅助栈现栈顶元素又为min。

所以辅助栈中存放的不只有一个min元素，而是主栈中较小元素的降序，如图：

具体实现思路：

用两个栈实现，栈 a 和栈 b
a 正常出入栈，b 的栈顶存放 a 中最小的元素
a 的push和pop操作都要维护 b 中栈顶是 a 中最小元素
对 a push x，同时判断 b 是否有元素，没有就也入栈 b；有就和 b 栈顶元素比较，x 小于 b 栈顶元素就入栈 b
对 a pop x，如果 b 栈顶的元素和 x 相等，则同时出栈
对 a top，b 不做任何处理

代码实现

核心方法就在push和pop，需要维护两个栈数据一致。

需要注意，栈 a 中是可能有重复元素，如果多个元素相等且是最小值，需要都压入栈 b ，这样后续对栈 a 的pop 操作，a 和 b 数据才能保持一致。

public class MinStack {
    LinkedList<Integer> a,b;
    public MinStack() {
        a = new LinkedList<Integer>();
        b = new LinkedList<Integer>();
    }

    public void push(int x) {
        a.addLast(x);
        if (b.isEmpty()) {
            b.addLast(x);
            return;
        }
        // >=  等于很重要，因为有可能两个元素相等
        // 如果b少入一次栈的话，可能就导致ab数据不一致了
        if (b.getLast() >= x) {
            b.addLast(x);
        }
    }

    public void pop() {
        int x = a.removeLast();
        if (b.getLast() == x) {
            b.removeLast();
        }
    }

    public int top() {
        return a.getLast();
    }

    public int min() {
        return b.getLast();
    }

    public static void main(String[] args) {
        MinStack minStack = new MinStack();
        minStack.push(10);
        minStack.push(2);
        minStack.push(3);
        System.out.println(minStack.min());
    }
}