设计一个具有getMin()功能的栈

最新推荐文章于 2024-10-16 18:40:41 发布

原创最新推荐文章于 2024-10-16 18:40:41 发布 · 384 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈

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

7 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方法，该栈支持push、pop及getMin操作的时间复杂度均为O(1)。通过维护两个栈实现快速获取最小元素的功能。

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

设计一个具有getMin()功能的栈：
1.push、pop、getMin()时间复杂度为O(1)
2.设计栈时可以使用现有的栈结构

/**
 * 设计一个具有getMin()功能的栈：
 * 1.pop，push，getMin()时间复杂度为O(1)
 * 2.设计的栈可以使用现有的栈结构
 * @author Nemo
 *
 */
public class StackDemo {

    private  Stack<Integer> stackData = new Stack<Integer>();
    private  Stack<Integer> stackMin = new Stack<Integer>();
    public void push(Integer data){
        if(data == null){
            throw new IllegalArgumentException("Illegal argument");
        }
        if(stackMin.isEmpty() || stackMin.peek() >=data){
            this.stackMin.push(data);
        }else{
            this.stackMin.push(stackMin.peek());
        }
        this.stackData.push(data);
    }

    public Integer pop(){
        if(stackData.isEmpty()){
            throw new RuntimeException("stack is empty");
        }
        this.stackMin.pop();
        return this.stackData.pop();
    }

    public Integer getMin(){
        if(stackMin.isEmpty()){
            throw new RuntimeException("stack is empty");
        }
        return stackMin.peek();
    }