【剑指 Offer】30. 包含 min 函数的栈

使用双栈实现最小值栈功能

原创于 2021-05-08 14:52:05 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #栈 #数据结构 #剑指offer

剑指offer 专栏收录该内容

9 篇文章

订阅专栏

这篇博客介绍了一个使用两个栈实现最小值栈的方法。通过维护一个额外的栈`s2`来保存每次遇到的最小值，可以在常数时间内完成查询最小值和常规的栈操作。Java代码展示了如何在push、pop、top和min操作中保持最小值栈的正确性。

原题链接

解析

定义两个栈，一个存储原来的元素，一个存储最小的元素

参考代码（Java）

class MinStack {

    Stack<Integer> s1, s2;

    public MinStack() {
        s1 = new Stack<>();
        s2 = new Stack<>();
    }

    public void push(int x) {
        s1.push(x);
        if (s2.isEmpty()) {
            s2.push(x);
        } else {
            s2.push(s2.peek() > x ? x : s2.peek());
        }
    }

    public void pop() {
        s1.pop();
        s2.pop();
    }

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

    public int min() {
        return s2.peek();
    }
}