剑指Offer 020包含min函数的栈

最新推荐文章于 2022-03-18 21:51:16 发布

原创最新推荐文章于 2022-03-18 21:51:16 发布 · 99 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

65 篇文章

订阅专栏

博客围绕实现含min函数的栈展开，要求min函数时间复杂度为O(1)。采用辅助栈法，创建两个栈，一个存数据，一个存最小值。详细介绍了push、pop、top、min操作的实现思路，还提及了代码实现但未给出具体代码。

020包含min函数的栈
题目：
定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））

方法：辅助栈法
思路：创建两个栈，一个用来存数据，一个用来存最小值：
push：data栈直接push，判断min栈是否为空，为空直接push，不为空，判断要插入的元素是否小于最上面的元素，小于就插入。
pop：判断如果data栈的最顶部元素跟min栈的最顶部元素一样就把两个栈的最顶部元素弄出，如果不相同，就只弄出data的最顶部元素
top：如果两个栈最顶部值一样，就直接返回这个值，pop出最顶部元素
min：判断min栈是否为空，为空就报错，不为空就pop出min栈最顶部元素

代码：

Stack<Integer> data = new Stack<>();
    Stack<Integer> min = new Stack<>();
    public void push(int node) {
        data.push(node);
        if(min.empty()){
            min.push(node);
        }else if(node<min.peek()){
            min.push(node);
        }
    }

    public void pop() {
        if(data.empty()){
            throw new RuntimeException("stack is empty");
        }else if(data.peek()==min.peek()){
            min.pop();
        }
        data.pop();
    }

    public int top() {
        if(data.empty()){
            throw new RuntimeException("stack is empty");
        }
        int num = data.pop();
        if(num==min.peek()){
            data.pop();
        }
        return num;
    }

    public int min() {
        if(min.empty()){
            throw new RuntimeException("stack is empty");
        }else{
            return min.peek();
        }
}