Java中PriorityQueue实现最小堆和最大堆的用法


Posted in Java/Android onJune 27, 2021

一、基本介绍

 1、介绍

学习很多算法知识,力争做到最优解的学习过程中,很多时候都会遇到PriorityQueue(优先队列)。一个基于优先级堆的无界优先级队列。优先级队列的元素按照其自然顺序进行排序,或者根据构造队列时提供的 Comparator 进行排序,具体取决于所使用的构造方法。优先级队列不允许使用 null 元素。依靠自然顺序的优先级队列还不允许插入不可比较的对象,这样做可能导致 ClassCastException。

此队列的头是按指定排序方式确定的最小元素。如果多个元素都是最小值,则头是其中一个元素——选择方法是任意的。队列获取操作 poll、remove、peek 和 element 访问处于队列头的元素。优先级队列是无界的,但是有一个内部容量,控制着用于存储队列元素的数组大小。它通常至少等于队列的大小。随着不断向优先级队列添加元素,其容量会自动增加。无需指定容量增加策略的细节。

此类及其迭代器实现了Collection和Iterator接口的所有可选方法。方法 iterator() 中提供的迭代器不保证以任何特定的顺序遍历优先级队列中的元素。如果需要按顺序遍历,请考虑使用 Arrays.sort(pq.toArray())。此实现不是同步的,如果多个线程中的任意线程修改了队列,则这些线程不应同时访问PriorityQueue实例。相反,请使用线程安全的PriorityBlockingQueue 类。

PriorityQueue翻译为优先队列,“优先”指元素在队列中按一定的顺序(优先级)进行存放,“队列”指一种先进先出的数据结构。因此PriorityQueue可以实现按照一定的优先级存取元素。

Java中PriorityQueue实现最小堆和最大堆的用法

2、用法

从源码来看PriorityQueue的构造方法:

//默认容量为 11
private static final int DEFAULT_INITIAL_CAPACITY = 11;
//1、无参构造,默认容量和默认排序方法
public PriorityQueue() {
        this(DEFAULT_INITIAL_CAPACITY, null);
    }
//2、指定容量
public PriorityQueue(int initialCapacity) {
        this(initialCapacity, null);
    }
//3、指定排序方法
public PriorityQueue(Comparator<? super E> comparator) {
        this(DEFAULT_INITIAL_CAPACITY, comparator);
    }
//4、指定容量和排序方法
public PriorityQueue(int initialCapacity,
                         Comparator<? super E> comparator) {
        // Note: This restriction of at least one is not actually needed,
        // but continues for 1.5 compatibility
        if (initialCapacity < 1)
            throw new IllegalArgumentException();
        this.queue = new Object[initialCapacity];
        this.comparator = comparator;
    }

由上可知,在构造PriorityQueue时我们可以指定初始容量和元素在队列中的排序方法,若不指定,则默认初始容量为11,默认排序方法为将元素从小到大进行排序。

3、最小堆

构造最小堆:

PriorityQueue<Integer> minheap = new PriorityQueue<>();

使用无参构造,元素在队列中默认按照从小到大的顺序排列,可保证每次出队列的元素为队列中的最小元素。

4、最大堆

PriorityQueue<Integer> maxheap = new PriorityQueue<>(Collections.reverseOrder());

将排序方法指定为反序,即元素从大到小排列,可保证每次出队列的元素为队列中最大的元素。

5、其他优先级

按照其他优先级规则排序,需要自己实现Comparable接口,重写compareTo()方法。

Comparable<Integer> comparable = new Comparable<Integer>() {
            @Override
            public int compareTo(Integer o) {
                return 0;
            }
        };

二、常用方法

以Integer类型为例:

Java中PriorityQueue实现最小堆和最大堆的用法

三、相关练习题

剑指 Offer 40. 最小的k个数

输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。

示例 1:

输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]

示例 2:

输入:arr = [0,1,2,1], k = 1
输出:[0]

限制:

0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

【解题思想】

先将k个数放进最大堆,再从第k+1个数开始比较,若其小于大堆顶则加入堆,堆顶出队列,若大于等于则无作为。

【代码】

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int res[] = new int[k];
        int len = arr.length;
        if(len == 0 || k == 0)
            return res;

        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Collections.reverseOrder());
        for(int i = 0; i < k; i++){
            maxHeap.add(arr[i]);
        }
        for(int i = k; i < len; i++){
            if(arr[i] < maxHeap.peek()){
                maxHeap.add(arr[i]);
                maxHeap.poll();
            }
        }
        
        for(int i = 0; i < k; i++){
            res[i] = maxHeap.poll();
        }
        return res;
    }
    
}

时间复杂度:O(nlogn)

到此这篇关于Java中PriorityQueue实现最小堆和最大堆的用法的文章就介绍到这了,更多相关Java PriorityQueue最小最大堆内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Java/Android 相关文章推荐
JVM入门之类加载与字节码技术(类加载与类的加载器)
Jun 15 Java/Android
Feign调用传输文件异常的解决
Jun 24 Java/Android
SpringBoot实现异步事件驱动的方法
Jun 28 Java/Android
死磕 java同步系列之synchronized解析
Jun 28 Java/Android
Java基础之线程锁相关知识总结
Jun 30 Java/Android
Spring mvc是如何实现与数据库的前后端的连接操作的?
Jun 30 Java/Android
java实现对Hadoop的操作
Jul 01 Java/Android
Java反应式框架Reactor中的Mono和Flux
Jul 25 Java/Android
聊聊SpringBoot自动装配的魔力
Nov 17 Java/Android
Spring Security使用单点登录的权限功能
Apr 03 Java/Android
详解Spring Bean的配置方式与实例化
Jun 10 Java/Android
详解Spring Security如何在权限中使用通配符
Jun 28 Java/Android
探讨Java中的深浅拷贝问题
解决SpringBoot跨域的三种方式
Jun 26 #Java/Android
分析Java中Map的遍历性能问题
Jun 26 #Java/Android
SpringCloud的JPA连接PostgreSql的教程
spring项目中切面及AOP的使用方法
Java 中的 Unsafe 魔法类的作用大全
Jun 26 #Java/Android
详解Java线程池是如何重复利用空闲线程的
Jun 26 #Java/Android
You might like
php实现快速排序的三种方法分享
2014/03/12 PHP
推荐几款用 Sublime Text 开发 Laravel 所用到的插件
2014/10/30 PHP
apache中为php 设置虚拟目录
2014/12/17 PHP
smarty模板引擎中变量及变量修饰器用法实例
2015/01/22 PHP
php实现统计目录文件大小的函数
2015/12/25 PHP
PHP设计模式之工厂模式与单例模式
2016/09/28 PHP
Laravel接收前端ajax传来的数据的实例代码
2017/07/20 PHP
在Laravel中使用MongoDB的方法示例
2019/11/11 PHP
jquery 简单导航实现代码
2009/09/11 Javascript
jQuery中获取Radio元素值的方法
2013/07/02 Javascript
JavaScript判断一个字符串是否包含指定子字符串的方法
2015/03/18 Javascript
jquery控制表单输入框显示默认值的方法
2015/05/22 Javascript
JS基于ocanvas插件实现的简单画板效果代码(附demo源码下载)
2016/04/05 Javascript
jQuery实现调整表格单列顺序完整实例
2016/06/20 Javascript
Javascript(es2016) import和require用法和区别详解
2017/08/11 Javascript
基于vue循环列表时点击跳转页面的方法
2018/08/31 Javascript
vue-cli4使用全局less文件中的变量配置操作
2020/10/21 Javascript
[01:24:09]Ti4 冒泡赛第二轮DK vs C9 1
2014/07/14 DOTA
浅谈利用numpy对矩阵进行归一化处理的方法
2018/07/11 Python
Python 利用pydub库操作音频文件的方法
2019/01/09 Python
python读csv文件时指定行为表头或无表头的方法
2019/06/26 Python
详解python 中in 的 用法
2019/12/12 Python
Python3中的f-Strings增强版字符串格式化方法
2020/03/04 Python
python应用Axes3D绘图(批量梯度下降算法)
2020/03/25 Python
python 使用raw socket进行TCP SYN扫描实例
2020/05/05 Python
Python爬虫小例子——爬取51job发布的工作职位
2020/07/10 Python
详解python3类型注释annotations实用案例
2021/01/20 Python
预备党员承诺书
2014/03/25 职场文书
如何写早恋检讨书
2014/09/10 职场文书
党员学习中共十八大报告思想汇报
2014/09/15 职场文书
乡镇党的群众路线教育实践活动制度建设计划
2014/11/03 职场文书
幼儿园大班开学寄语(2015秋季)
2015/05/27 职场文书
《搭石》教学反思
2016/02/18 职场文书
sql查询结果列拼接成逗号分隔的字符串方法
2021/05/25 SQL Server
win10电脑老是死机怎么办?win10系统老是死机的解决方法
2022/08/05 数码科技
Nginx 502 bad gateway错误解决的九种方案及原因
2022/08/14 Servers