定义栈结构实现min(),push(),pop()函数要求时间复杂度为O(1)

原创于 2021-11-06 18:49:23 发布

· 289 阅读

0 ·

版权

文章标签：

#java #栈 #算法

简单算法的核心思想专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何定义一个栈结构，使得min()、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,b;
    public MinStack() {
        a = new Stack();
        b = new Stack();
    }
    
    public void push(int x) {
        if(b.empty() || b.peek() >= x){
            b.push(x);
        }
        a.push(x);
    }
    
    public void pop() {
        if(a.pop().equals(b.peek())){
            b.pop();
        }
    }
    
    public int top() {
        return a.peek();
    }
    
    public int min() {
        return b.peek();
    }
}