Java 优先队列（PriorityQueue）的使用

最新推荐文章于 2025-10-16 20:57:26 发布

原创最新推荐文章于 2025-10-16 20:57:26 发布 · 1.6w 阅读

36 ·

CC 4.0 BY-SA版权

JavaSE 专栏收录该内容

11 篇文章

订阅专栏

本文介绍使用小顶堆实现优先队列的基本方法，包括add、element/peek、remove/poll等常用操作，并通过示例代码展示了如何在Java中实现这一数据结构。

常用方法

1.add(e)

2.element()、peek()

element()和peek()的语义完全相同，都是获取但不删除队首元素，也就是队列中权值最小的那个元素，二者唯一的区别是当方法失败时前者抛出异常，后者返回null。根据小顶堆的性质，堆顶那个元素就是全局最小的那个；由于堆用数组表示，根据下标关系，0下标处的那个元素既是堆顶元素。所以直接返回数组0下标处的那个元素即可。

3.remove()、poll()

remove()和poll()方法的语义也完全相同，都是获取并删除队首元素，区别是当方法失败时前者抛出异常，后者返回null。由于删除操作会改变队列的结构，为维护小顶堆的性质，需要进行必要的调整。

示例代码

import java.util.*;
class Node 
{
	public int key;
	public int sec;
}
public class Main
{
	static Node []A = null;
	public static void main(String args[]) 
	{
		Scanner in = new Scanner(System.in);
		Comparator<Node> cmp = new Comparator<Node>() {
			public int compare(Node a, Node b) { //这里是小根堆
				if (a.key == b.key) return a.sec - b.sec; //第二键值做参考
				return a.key - b.key; //第一键值做参考
			}
		};
		Queue<Node> pq = new PriorityQueue<>(cmp);
		int n = in.nextInt();
		A = new Node [n];
		for (int i = 0; i < n; i++) {
			A[i] = new Node();
			A[i].key = in.nextInt();
			A[i].sec = in.nextInt();
			pq.add(A[i]);
		}
		while (!pq.isEmpty()) {
			System.out.println(pq.element().key + " " + pq.element().sec);
			pq.poll();
		}
	}
}