登录
首页 >  文章 >  java教程

如何在Java中使用PriorityQueue优先队列

时间:2025-10-16 23:27:11 238浏览 收藏

在文章实战开发的过程中,我们经常会遇到一些这样那样的问题,然后要卡好半天,等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《如何在Java中使用PriorityQueue优先队列》,聊聊,希望可以帮助到正在努力赚钱的你。

PriorityQueue是Java中基于堆的优先队列,默认小顶堆,可通过Comparator自定义排序,常用于贪心算法、Dijkstra等场景,支持offer、poll、peek等操作,非线程安全且不支持null元素。

如何在Java中使用PriorityQueue优先队列

在Java中,PriorityQueue 是一个基于堆的数据结构,可以自动对元素进行排序,保证每次取出的元素都是优先级最高的(默认最小值)。它常用于实现贪心算法、Dijkstra最短路径、合并K个有序链表等场景。

1. 基本使用方法

PriorityQueue位于 java.util 包中,使用前需要导入。默认情况下,它实现的是小顶堆,即队首元素是最小的。

创建一个存储整数的优先队列:

PriorityQueue pq = new PriorityQueue(); pq.offer(5); pq.offer(1); pq.offer(3); System.out.println(pq.poll()); // 输出 1 System.out.println(pq.poll()); // 输出 3 System.out.println(pq.poll()); // 输出 5

注意:poll() 方法取出并移除队首元素;peek() 只查看不移除。

2. 自定义比较器(大顶堆或对象排序)

如果想让优先队列变成大顶堆(最大值优先),可以通过传入自定义的 Comparator 实现。

PriorityQueue maxPQ = new PriorityQueue((a, b) -> b - a); maxPQ.offer(5); maxPQ.offer(1); maxPQ.offer(3); System.out.println(maxPQ.poll()); // 输出 5

也可以使用 Collections.reverseOrder() 快速创建大顶堆:

PriorityQueue maxPQ = new PriorityQueue(Collections.reverseOrder());

对于自定义对象,比如按学生的分数排序:

class Student { String name; int score; Student(String name, int score) { this.name = name; this.score = score; } } PriorityQueue students = new PriorityQueue((a, b) -> b.score - a.score); students.offer(new Student("Alice", 85)); students.offer(new Student("Bob", 90)); System.out.println(students.peek().name); // 输出 Bob

3. 常用方法说明

  • offer(E e):添加元素,失败时抛出异常(队列满时一般不会发生,因为PriorityQueue是无界的)
  • poll():取出并删除队首元素,队列为空返回 null
  • peek():查看队首元素但不删除,为空返回 null
  • size():返回当前元素个数
  • isEmpty():判断是否为空

示例:

while (!pq.isEmpty()) { System.out.println(pq.poll()); }

4. 注意事项

PriorityQueue 不是线程安全的。如果需要在多线程环境中使用,考虑使用 PriorityBlockingQueue

它不允许插入 null 元素,否则会抛出 NullPointerException。

迭代遍历时顺序不一定有序,因为内部只是保证堆结构,并非完全排序数组。要获取有序数据,应连续调用 poll()。

基本上就这些。掌握构造、比较器和基本操作就能应对大多数场景了。

理论要掌握,实操不能落!以上关于《如何在Java中使用PriorityQueue优先队列》的详细介绍,大家都掌握了吧!如果想要继续提升自己的能力,那么就来关注golang学习网公众号吧!

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>