Java数据结构之堆(优先队列)


Posted in Java/Android onMay 20, 2022

堆(优先队列)是一种典型的数据结构,其形状是一棵完全二叉树,一般用于求解topk问题。根据双亲节点大于等于孩子节点或双亲节点小于等于孩子节点,可分为大顶堆和小顶堆,本文实现大顶堆。

根据大顶堆的定义,大顶堆的双亲节点大于等于其孩子节点,堆顶元素最大,对于每一个子树都是一个大顶堆,则从最后一个双亲节点进行调整为大顶堆,一直到根节点,则可构建一个大顶堆。

我们这里采用数组去存储,以heap={3,2,1,5,6,4}为例,需要一个init(int[] heap)初始化方法,从最后一个双亲节点开始将heap逐渐调整为大顶堆,其中需要使用到adjust(int[] heap, int i, int end)方法。

调整过程:从最后一个双亲节点出发,如果以当前双亲节点为根的树不符合大顶堆,则进行调整。

Java数据结构之堆(优先队列)

代码实现如下:

public void init(int[] heap) {
        //从最后一个双亲节点开始调整
        //逐渐往上进行调整
        for (int i = heap.length / 2 ; i > 0 ; i-- ) {
            this.adjust(heap, i, heap.length);
        }
    }

    public void adjust(int[] heap, int i, int end) {
        int j = i << 1;
        while (j <= end) {
            //找到两个孩子节点z中较大的节点
            if (j < end && heap[j - 1] < heap[j]) {
                j = j + 1;
            }
            //如果较大节点还小于根节点,则以当前节点为根节点的
            //二叉树已经是大顶堆,不需要进行调整
            if (heap[i - 1] > heap[j - 1]) {
                break;
            }
            //进行调整,将当前节点换到较大位置,再从当前位置进行调整
            int temp = heap[i - 1];
            heap[i - 1] = heap[j - 1];
            heap[j - 1] = temp;
            i = j;
            j = i << 1;
        }
    }

构建好了大顶堆之后,我们如何求得topk呢,此时堆顶元素为top1,我们只需要将top1元素拿走,将剩下元素调整为大顶堆,k次之后即可得到topk。

具体过程:我们将堆顶元素与最后一个元素进行交换,然后将堆顶到倒数第二个元素进行调整,依次类推。

Java数据结构之堆(优先队列)

以leetcode215数组中第k个最大元素为例:

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素

public int findKthLargest(int[] nums, int k) {
        this.init(nums);
        //找到第k大的数
        int end = nums.length;
        while (k > 1) {
            //将当前堆顶元素放到末尾,进行堆调整
            int temp = nums[0];
             nums[0] = nums[end - 1];
             nums[end - 1] = temp;
             end = end - 1;
             -- k;
             this.adjust(nums, 1, end);
        }
        return nums[0];
    }

Java数据结构之堆(优先队列)

此外,Java本身提供了优先队列集合类,但是对于这个题目效率不如自己实现的高

public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k);
        for (int num : nums) {
            if (priorityQueue.size() == k) {
                if (num > priorityQueue.peek()) {
                    priorityQueue.poll();
                    priorityQueue.add(num);
                }
                continue;
            }
            priorityQueue.add(num);
        }
        return priorityQueue.poll();
    }

Java数据结构之堆(优先队列)

到此这篇关于Java数据结构之堆(优先队列)的实现的文章就介绍到这了,更多相关Java 堆内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!


Tags in this post...

Java/Android 相关文章推荐
Spring Bean的实例化之属性注入源码剖析过程
Jun 13 Java/Android
分析Netty直接内存原理及应用
Jun 14 Java/Android
一篇带你入门Java垃圾回收器
Jun 16 Java/Android
Feign调用全局异常处理解决方案
Jun 24 Java/Android
idea 在springboot中使用lombok插件的方法
Aug 02 Java/Android
JavaCV实现照片马赛克效果
Jan 22 Java/Android
Java实现学生管理系统(IO版)
Feb 24 Java/Android
剑指Offer之Java算法习题精讲二叉树的构造和遍历
Mar 21 Java/Android
Spring依赖注入多种类型数据的示例代码
Mar 31 Java/Android
Java实现注册登录跳转
Jun 16 Java/Android
springboot+rabbitmq实现智能家居实例详解
Jul 23 Java/Android
Java中的Kafka为什么性能这么快及4大核心详析
Sep 23 Java/Android
Java中Dijkstra(迪杰斯特拉)算法
Android Studio实现带三角函数对数运算功能的高级计算器
May 20 #Java/Android
springcloud整合seata
springboot读取nacos配置文件
May 20 #Java/Android
Android studio 简单计算器的编写
May 20 #Java/Android
mybatis 获取更新记录的id
May 20 #Java/Android
Android Studio 计算器开发
May 20 #Java/Android
You might like
如何给phpadmin一个保护
2006/10/09 PHP
基于Jquery的标签智能验证实现代码
2010/12/27 Javascript
javascript 另一种图片滚动切换效果思路
2012/04/20 Javascript
javascipt匹配单行和多行注释的正则表达式
2013/11/20 Javascript
js与jquery获取父元素,删除子元素的两种不同方法
2014/01/09 Javascript
jQuery实现两款有动画功能的导航菜单代码
2015/09/16 Javascript
JS实现兼容性好,自动置顶的淘宝悬浮工具栏效果
2015/09/18 Javascript
javascript实现计时器的简单方法
2016/02/21 Javascript
JavaScript计时器用法分析【setTimeout和clearTimeout】
2017/01/18 Javascript
Move.js入门
2017/02/08 Javascript
jQuery表单设置值的方法
2017/06/30 jQuery
react路由配置方式详解
2017/08/07 Javascript
jquery实现简单拖拽效果
2020/07/20 jQuery
python调用机器喇叭发出蜂鸣声(Beep)的方法
2015/03/23 Python
分享给Python新手们的几道简单练习题
2017/09/21 Python
python: 自动安装缺失库文件的方法
2018/10/22 Python
Python3转换html到pdf的不同解决方案
2019/03/11 Python
python制作抖音代码舞
2019/04/07 Python
Python3 列表,数组,矩阵的相互转换的方法示例
2019/08/05 Python
Pytorch模型转onnx模型实例
2020/01/15 Python
canvas进阶之如何画出平滑的曲线
2018/10/15 HTML / CSS
布鲁明戴尔百货店:Bloomingdale’s
2016/12/21 全球购物
美国Lolё官网:购买大胆而美丽的女性运动服装
2017/05/22 全球购物
计算 s=(x*y)1/2,用两个宏定义来实现
2016/08/11 面试题
临床医学专业学生的自我评价分享
2013/11/21 职场文书
《伯牙绝弦》教学反思
2014/03/02 职场文书
公司接待方案
2014/03/08 职场文书
厂区绿化方案
2014/05/08 职场文书
美术课外活动总结
2014/07/08 职场文书
2014年精神文明工作总结
2014/12/23 职场文书
白鹤梁导游词
2015/02/06 职场文书
2015年反腐倡廉工作总结
2015/05/14 职场文书
三十年再续同学情倡议书
2019/11/27 职场文书
Python Pycharm虚拟下百度飞浆PaddleX安装报错问题及处理方法(亲测100%有效)
2021/05/24 Python
如何正确理解python装饰器
2021/06/15 Python
Golang流模式之grpc的四种数据流
2022/04/13 Golang