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

最新推荐文章于 2023-01-28 16:03:14 发布

MGfanfan

最新推荐文章于 2023-01-28 16:03:14 发布

阅读量57

点赞数

分类专栏：剑指Offer 文章标签： java 栈

本文链接：https://blog.csdn.net/MGfanfan/article/details/112512398

版权

剑指Offer 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

该博客介绍了如何通过维护一个辅助栈来实现在push、pop、top操作的同时，min函数的时间复杂度为O(1)。在主栈A存储所有元素，辅助栈B则始终保存栈A中的最小值，当元素小于B栈顶时才入B栈。这样，top和min操作只需查询相应栈的栈顶，而pop时比较A栈顶元素与B栈顶元素，相等则同时出栈。这种方法保证了在20000次操作内高效找到栈的最小元素。

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

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 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.

提示：

各函数的调用总次数不超过 20000 次

分析

将 min() 函数复杂度降为 O(1)，可通过建立辅助栈实现；
栈A：输入的数据
栈B：输入数据的一种降序，就是为了找出栈A中最小的值

1.push函数

将x压入栈内A（A.add(x）)
B栈的情况是当B为空或者x<=栈B的栈顶元素时，将x压入B内

2.pop函数：保持栈A\B的元素一致性

执行栈 AA 出栈（即 A.pop() ），将出栈元素记为 yy ；
若 yy 等于栈 B 的栈顶元素，则执行栈 B 出栈（即 B.pop() ）

3.top() 函数：直接返回栈 A 的栈顶元素即可，即返回 A.peek() 。

4.min() 函数：直接返回栈 BB 的栈顶元素即可，即返回 B.peek() 。

代码

class MinStack {
Stack<Integer> A,B;
    /** initialize your data structure here. */
    public MinStack() {
        A= new Stack<>();
        B= new Stack<>();

    }
    
    public void push(int x) {
        A.add(x);
        if(B.empty() || B.peek()>=x)//将A栈中小于B栈栈顶的元素加入到B栈，确保B栈始终降序
            B.add(x);
        
    }
    
    public void pop() {
        if(A.pop().equals(B.peek()))
        B.pop();

    }
    
    public int top() {
        return A.peek();
    }
    
    public int min() {
        return B.peek();//返回B栈中的栈顶元素

    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.min();
 */