讲师中心 微信公众号
AI工具推荐 视频效率加速

Java中PriorityQueue的核心使用技巧

大浩小哥_7956

大浩小哥_7956

发布时间:2025-09-22 12:40:01

|

603人浏览过

|

来源于php中文网

原创

答案:PriorityQueue是基于堆的优先级队列,默认为最小堆,用于高效获取极值,适用于任务调度、Top K等问题;它不保证全局有序,允许重复元素,添加和删除操作时间复杂度为O(log N),但remove(Object)效率低,遍历时无序,需用poll()按序取出;自定义排序可通过Comparator实现,如最大堆或对象字段排序;注意初始容量设置、不可变排序字段、非线程安全及禁止null元素,与TreeSet相比更适用于仅关注极值的场景。

java中priorityqueue的核心使用技巧

Java中的PriorityQueue,在我看来,它就是那个能帮你高效处理“谁最重要”这类问题的工具。它本质上是一个基于堆(heap)数据结构的优先级队列,默认情况下,它会帮你把最小的那个元素放在队头,让你能以非常快的速度拿到它。如果你需要一个总是能告诉你“当前最小(或最大)值是什么”的集合,并且你主要关心的是快速获取这个极值,而不是遍历整个集合的排序,那么PriorityQueue就是你的理想选择。它不是一个排序列表,而是一个为你维护优先级顺序的队列。

PriorityQueue的核心使用,其实围绕着几个基本操作展开,但理解它们背后的逻辑至关重要。

当你需要创建一个PriorityQueue时,通常有两种方式:

  1. 默认最小堆:
    PriorityQueue<Integer> minHeap = new PriorityQueue<>();
    这种情况下,它会根据元素的自然顺序(比如整数从小到大)来排序。
  2. 自定义排序:
    PriorityQueue<String> maxHeap = new PriorityQueue<>((s1, s2) -> s2.length() - s1.length());
    这里我们传入一个Comparator,决定了元素如何比较。比如,这个例子就是根据字符串长度从长到短排序,模拟一个最大堆。

往队列里添加元素,你可以用

add()
或
offer()
,它们的功能几乎一样,都是将元素插入到正确的位置,确保堆的性质得到维护。这个操作的时间复杂度是O(log N),效率很高。

立即学习“Java免费学习笔记(深入)”;

获取元素时,

peek()
会返回队头元素(也就是优先级最高的那个,默认是最小的),但不会将其移除。而
poll()
则会返回并移除队头元素。这两个操作也都是O(log N)。

一个常见的误解是,很多人以为PriorityQueue内部是完全排序的,比如你用迭代器去遍历它,会得到一个有序的序列。但实际上,这是不对的。PriorityQueue只保证队头元素是最小(或最大)的,内部其他元素的顺序只满足堆的性质,并不保证全局有序。如果你真的需要一个完全排序的集合,TreeSet可能更适合你。

PriorityQueue和TreeSet有什么区别,我什么时候该用它?

这个问题我经常被问到,它们俩确实有些相似之处,但核心用途大相径庭。简单来说,PriorityQueue是为了“快速找到并处理优先级最高的元素”而生的,而TreeSet则是为了“维护一个始终有序且不重复的集合”而存在。

PriorityQueue:

  • 关注点: 极值(最小或最大)的快速存取。
  • 内部结构: 堆(通常是二叉堆)。
  • 元素重复: 允许重复元素。如果你添加两个相同的整数,它们都会在队列中。
  • 迭代顺序: 不保证迭代器遍历时是完全有序的,只保证
    poll()
    出来的元素是有序的。
  • 典型场景: 任务调度(优先级高的任务先执行)、Dijkstra算法(查找最短路径中的下一个节点)、Top K问题(找出最大或最小的K个元素)、模拟事件队列。

TreeSet:

  • 关注点: 维护一个有序的、不重复的元素集合。
  • 内部结构: 红黑树(一种自平衡二叉查找树)。
  • 元素重复: 不允许重复元素。如果你添加一个已经存在的元素,它会被忽略。
  • 迭代顺序: 迭代器遍历时,元素是严格按照自然顺序或自定义顺序排列的。
  • 典型场景: 需要一个始终保持排序的集合、需要高效地进行范围查询(例如,找出所有大于X小于Y的元素)、维护唯一且有序的ID列表。

所以,如果你只是想快速拿到最小或最大的那个,并且不关心其他元素的全局排序,PriorityQueue是更高效的选择。但如果你需要一个随时都是有序的、且元素唯一的集合,甚至需要进行范围查询,那么TreeSet会更合适。举个例子,如果我要实现一个简单的任务调度器,每次都取出优先级最高的任务来执行,PriorityQueue就是我的首选。但如果我要维护一个用户在线列表,并且需要随时能按用户名排序显示,TreeSet就更合适。

如何自定义PriorityQueue的排序规则?

自定义PriorityQueue的排序规则,是它变得真正强大的地方。因为默认的自然排序很多时候并不能满足我们的需求。这主要通过在构造函数中传入一个

Comparator
来实现。

最常见的场景就是把默认的最小堆变成最大堆。比如,如果你想让整数从大到小排列:

Java注释规范Skills
Java注释规范Skills

当用户要求为 Java 代码添加注释、修改注释、审查注释、整理注释格式,或要求按照代码注释规范处理 Java 文件时,使用本技能。本技能用于统一生成和维护 Java 类注释模板、按 Git 对比 release 判断版本号、补充每行代码上方中文块注释、调整注释空行规则,并保持 Java 文件 UTF-8 无 BOM 编码。典型触发场景包括:补充类注释、更新 @version、检查 @author 来源、整理行尾注释、把注释放到代码上一行、检查注释是否符合规范、构建前检查 Java 注释风格。

下载
// 方法一:匿名内部类(传统方式)
PriorityQueue<Integer> maxHeapTraditional = new PriorityQueue<>(new Comparator<Integer>() {
    @Override
    public int compare(Integer a, Integer b) {
        return b - a; // b - a 会让大的排在前面
    }
});

// 方法二:Lambda表达式(更简洁,Java 8+)
PriorityQueue<Integer> maxHeapLambda = new PriorityQueue<>((a, b) -> b - a);

这两种方式都实现了同样的效果:让大的整数拥有更高的优先级(即排在队头)。

如果你要处理自定义对象,比如一个

Task
类,它有一个
priority
字段,你希望根据这个字段来排序:

class Task {
    String name;
    int priority; // 优先级,数字越小优先级越高

    public Task(String name, int priority) {
        this.name = name;
        this.priority = priority;
    }

    public int getPriority() {
        return priority;
    }

    @Override
    public String toString() {
        return "Task{" + "name='" + name + '\'' + ", priority=" + priority + '}';
    }
}

// 根据Task的priority字段进行排序,priority值越小越优先(默认行为)
PriorityQueue<Task> taskQueue = new PriorityQueue<>(Comparator.comparingInt(Task::getPriority));

// 如果想让priority值越大越优先
PriorityQueue<Task> taskQueueReverse = new PriorityQueue<>(Comparator.comparingInt(Task::getPriority).reversed());

// 使用Lambda表达式直接比较
PriorityQueue<Task> customTaskQueue = new PriorityQueue<>((t1, t2) -> Integer.compare(t1.getPriority(), t2.getPriority()));

通过

Comparator.comparingInt(Task::getPriority)
这种方式,代码可读性会非常好。你也可以链式调用
reversed()
来反转排序顺序。

记住,如果你放入PriorityQueue的对象本身实现了

Comparable
接口,并且你没有提供
Comparator
,那么PriorityQueue会使用对象的自然顺序。但一旦你提供了
Comparator
,它就会覆盖对象的自然排序。这种灵活性使得PriorityQueue能够适应各种复杂的优先级需求。

PriorityQueue在使用中常见的性能陷阱和注意事项有哪些?

虽然PriorityQueue功能强大,但在实际使用中,如果不注意一些细节,可能会遇到性能问题或者逻辑错误。我个人在项目中就踩过几次坑,所以这些点特别值得强调。

  1. remove(Object)
    的低效性: 这是个大坑。如果你需要从PriorityQueue中移除一个特定元素(而不是队头元素),
    remove(Object)
    方法的效率可能非常低。因为它需要遍历整个队列来找到那个元素(O(N)),然后重新调整堆结构(O(log N)),所以最坏情况下是O(N)或者O(N log N)。如果你经常需要移除非队头元素,可能需要重新考虑数据结构的选择,或者维护一个额外的Map来快速定位元素,但这会增加复杂性。

  2. 迭代顺序不保证有序: 我之前提过,这里再强调一次。

    for-each
    循环遍历PriorityQueue时,元素的顺序是不确定的,它只是按照堆的内部结构来遍历,而不是按照优先级顺序。如果你需要按优先级顺序处理所有元素,正确的做法是反复调用
    poll()
    直到队列为空。

  3. 初始容量的选择:

    PriorityQueue
    默认的初始容量是11。如果你的队列会存储大量元素,并且你知道大概的数量,最好在构造时就指定一个较大的初始容量,例如
    new PriorityQueue<>(initialCapacity)
    。这样可以避免频繁的扩容操作,每次扩容都会涉及数组的复制,带来额外的性能开销。虽然Java的扩容机制相对高效,但能避免还是尽量避免。

  4. 可变对象的陷阱: 如果你把自定义对象放入PriorityQueue,并且这些对象的某些字段是用于排序的(比如

    Task
    的
    priority
    ),那么当这些字段在对象被添加到队列后发生改变时,PriorityQueue不会自动调整其内部结构。这意味着你的队列可能不再保持正确的优先级顺序。解决办法通常是,当对象的排序字段改变时,先从队列中
    remove()
    该对象,更新其字段,然后再
    add()
    回队列。这个操作虽然是O(N)的,但可以确保堆的正确性。

  5. 非线程安全:

    PriorityQueue
    不是线程安全的。如果在多线程环境下使用,并且有多个线程同时对它进行修改(添加、移除),你需要手动进行同步,或者使用Java并发包中提供的
    PriorityBlockingQueue
    。
    PriorityBlockingQueue
    在内部已经处理了线程安全问题,但它在性能上会有一些开销。

  6. 不允许

    null
    元素:
    PriorityQueue
    不允许存储
    null
    元素。如果你尝试添加
    null
    ,它会抛出
    NullPointerException
    。这是因为
    null
    无法进行比较,也无法确定其优先级。

理解并规避这些潜在的问题,能让你更有效地利用PriorityQueue的优势,避免不必要的麻烦。在我看来,掌握这些细节,才是真正掌握一个工具的标志。

热门AI工具

更多
豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

相关专题

更多
string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

6859

2023.08.02

c语言中null和NULL的区别
c语言中null和NULL的区别

c语言中null和NULL的区别是:null是C语言中的一个宏定义,通常用来表示一个空指针,可以用于初始化指针变量,或者在条件语句中判断指针是否为空;NULL是C语言中的一个预定义常量,通常用来表示一个空值,用于表示一个空的指针、空的指针数组或者空的结构体指针。

549

2023.09.22

java中null的用法
java中null的用法

在Java中,null表示一个引用类型的变量不指向任何对象。可以将null赋值给任何引用类型的变量,包括类、接口、数组、字符串等。想了解更多null的相关内容,可以阅读本专题下面的文章。

1698

2024.03.01

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

1818

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

2904

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

6204

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

6049

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

7654

2024.03.22

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

60

2026.10.28

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn