包含min函数的栈

最新推荐文章于 2023-07-24 10:28:56 发布

原创最新推荐文章于 2023-07-24 10:28:56 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

60 篇文章

订阅专栏

本文介绍了一种特殊的数据结构——包含min函数的栈，该栈在进行push、pop和min操作时均可达到O(1)的时间复杂度。通过使用辅助栈记录每次push操作时的最小值，从而在不增加额外时间复杂度的情况下实现了快速获取栈中最小元素的功能。


import java.util.Stack;

public class E30StackForMin<T extends Comparable<T>> {
    //实现一个包含min函数的栈，push，pop，min操作的时间复杂度都为O(1)
    //辅助栈顶始终保存当前栈高度对应的栈中最小值
    private Stack<T> storeStack = new Stack<>();
    private Stack<T> assistStack = new Stack<>();

    public void push(T type){
        storeStack.push(type);
        if (assistStack.empty() || type.compareTo(assistStack.peek()) < 0)
            assistStack.push(type);
        else
            assistStack.push(assistStack.peek());
    }

    public void pop(){
        if (assistStack.empty() || storeStack.empty())
            throw new IllegalArgumentException("Null Stack");
        storeStack.pop();
        assistStack.pop();
    }

    public T min(){
        if (assistStack.empty() || storeStack.empty())
            throw new IllegalArgumentException("Null Stack");
        return assistStack.peek();
    }

    //测试用例
    public static void main(String[] args){
        E30StackForMin<Integer> stack = new E30StackForMin<>();
        stack.push(1);
        stack.push(0);
        stack.push(3);
        stack.pop();
        stack.pop();
        System.out.print(stack.min());
    }
}