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 相关文章推荐
分享一些Java的常用工具
Jun 11 Java/Android
springboot利用redis、Redisson处理并发问题的操作
Jun 18 Java/Android
springboot如何初始化执行sql语句
Jun 22 Java/Android
Java 中的 Unsafe 魔法类的作用大全
Jun 26 Java/Android
小程序与后端Java接口交互实现HelloWorld入门
Jul 09 Java/Android
JavaWeb 入门:Hello Servlet
Jul 16 Java/Android
Spring this调用当前类方法无法拦截的示例代码
Mar 20 Java/Android
Java基础——Map集合
Apr 01 Java/Android
零基础学java之带返回值的方法的定义和调用
Apr 10 Java/Android
Ubuntu18.04下QT开发Android无法连接设备问题解决实现
Jun 01 Java/Android
java实现面板之间切换功能
Jun 10 Java/Android
Java 中的 Lambda List 转 Map 的多种方法详解
Jul 07 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
收音机指标测试方法及仪器
2021/03/01 无线电
Look And Say 序列php实现代码
2011/05/22 PHP
PHP中配置IIS7实现基本身份验证的方法
2015/09/24 PHP
php微信公众号开发之图片回复
2018/10/20 PHP
学习ExtJS(二) Button常用方法
2009/10/07 Javascript
JavaScript 拾碎[三] 使用className属性
2010/10/16 Javascript
javascript将数组插入到另一个数组中的代码
2013/01/10 Javascript
JS和jquery获取各种屏幕的宽度和高度的代码
2013/08/02 Javascript
JQuery设置文本框和密码框得到焦点时的样式
2013/08/30 Javascript
解析JavaScript中delete操作符不能删除的对象
2013/12/03 Javascript
9款2014最热门jQuery实用特效推荐
2014/12/07 Javascript
jQuery制作效果超棒的手风琴折叠菜单
2015/04/03 Javascript
javascript下拉列表菜单的实现方法
2015/11/18 Javascript
JavaScript  event对象整理及详细介绍
2016/10/10 Javascript
Jquery uploadify 多余的Get请求(404错误)的解决方法
2017/01/26 Javascript
详解Vue组件实现tips的总结
2017/11/01 Javascript
VUE2.0+Element-UI+Echarts封装的组件实例
2018/03/02 Javascript
layui当点击文本框时弹出选择框,显示选择内容的例子
2019/09/02 Javascript
从零学Python之入门(二)基本数据类型
2014/05/25 Python
一个超级简单的python web程序
2014/09/11 Python
深入理解 Python 中的多线程 新手必看
2016/11/20 Python
python:动态路由的Flask程序代码
2019/11/22 Python
后端开发使用pycharm的技巧(推荐)
2020/03/27 Python
python3中celery异步框架简单使用+守护进程方式启动
2021/01/20 Python
基于第一个PhoneGap(cordova)的应用详解
2013/05/03 HTML / CSS
从零实现一个自定义html5播放器的示例代码
2017/08/01 HTML / CSS
Jones Bootmaker官网:优质靴子和鞋子在线
2020/11/30 全球购物
颇特女士香港官网:NET-A-PORTER香港
2021/03/08 全球购物
应届生幼儿园求职信
2013/11/12 职场文书
学校百日安全生产活动总结
2014/07/05 职场文书
大学生个人求职信例文
2014/07/07 职场文书
十佳家长事迹材料
2014/08/26 职场文书
交通事故死亡赔偿协议书
2014/12/03 职场文书
党员观看《筑梦中国》心得体会
2016/01/18 职场文书
一篇文章弄清楚Ajax请求的五个步骤
2022/03/17 Javascript
在虚拟机中安装windows server 2008的图文教程
2022/06/28 Servers