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

最新推荐文章于 2024-08-16 23:08:28 发布

无掺杂

最新推荐文章于 2024-08-16 23:08:28 发布

阅读量249

点赞数 1

文章标签： java

本文链接：https://blog.csdn.net/weixin_53968665/article/details/121550384

版权

该博客介绍了一个LeetCode题目，目标是设计一个数据结构，使得在常数时间内完成push、pop操作的同时，还能获取栈中的最小元素。解决方案是通过维护两个栈，一个用于存储所有元素，另一个用于存储当前最小值。每次push时，如果新元素小于或等于栈B的栈顶元素，则将其放入栈B。pop时，如果弹出的元素与栈B的栈顶元素相同，则同时弹出B的栈顶元素。这样，栈顶元素始终是当前最小值。

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

题目链接：包含min函数的栈
力扣题目：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。
思路：

定义两个栈，一个栈A用来存放所有数据，一个栈B用来存放当前最小值。
第一次放数据x0（此时A==null,B==null），同时放入栈A，B。
第二次放数据x1，由于A是用来存放数据的，所以给A直接入栈即可，由于B中放最小元素，所以用要放的数据x1和B中第一次放的元素x0做比较，若x1>=x0，则将第二次数据x1入栈。（注：为什么B放最小元素，这里不用出栈第一次数据x0，而是直接将当前最小数x1入栈？因为栈是“先进后出”的，所以只需要保证顶部元素是当前最小即可。）
第三次放数据x2与上步一样。
最小栈弹出元素时要注意，弹出A中元素，当前最小值有可能变化。所以不是只出A中元素，应在出A栈顶元素同时与B中栈顶元素比较是否相等，若相等，则同时应将B栈顶元素也弹出。
最后，获取栈顶元素直接A.peek(); 获取最小值直接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.isEmpty()||B.peek()>=x){
            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();
    }
}

/**
 * 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();
 */