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

最新推荐文章于 2024-08-23 14:01:41 发布

眼红de熊熊

最新推荐文章于 2024-08-23 14:01:41 发布

阅读量107

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构与算法

本文链接：https://blog.youkuaiyun.com/JhonyLin/article/details/116561204

数据结构与算法专栏收录该内容

63 篇文章

订阅专栏

本文介绍了一种特殊栈的数据结构实现，该栈包含min函数可在O(1)时间复杂度内获取栈内最小值。通过两个栈A和B协同工作，确保了push、pop操作也保持O(1)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述
定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.

问题解决

class MinStack {
    /** initialize your data structure here. */
    private Stack<Integer> A;
    private Stack<Integer> B;
    public MinStack() {
        this.A = new Stack<>();
        this.B = new Stack<>();
    }   
    
    public void push(int x) {
        A.push(x);
        if(B.isEmpty() || x <= B.peek()) {
            B.push(x);
        }
    }
    public void pop() {
        if(A.isEmpty()) return;
        if(A.pop().equals(B.peek())) {
            B.pop();
        }
    }
    public int top() {
        return A.peek();
    }
    public int min() {
        return B.peek();
    }
}