java 优先队列用法

在Java 中，优先队列是一种基于堆的数据结构，通常用于实现具有优先级的队列。在优先队列中，元素按照它们的优先级被处理，而不是按照它们被插入的顺序。Java 中的PriorityQueue 类是一个标准的优先队列实现，它可以存储可比较的元素，并根据它们的比较顺序进行排列。

以下是Java 中PriorityQueue 的基本用法：

导入PriorityQueue 类：

import java.util.PriorityQueue;

创建一个优先队列：

PriorityQueue priorityQueue = new PriorityQueue<>();

其中，T 是要存储在队列中的元素类型。默认情况下，它会根据元素的自然顺序进行排序。如果要使用自定义的比较规则，可以提供一个Comparator 对象作为参数。

添加元素到优先队列：

priorityQueue.add(element); // 或priorityQueue.offer(element);

这将把元素插入到优先队列中，并根据优先级对其进行排序。

获取队列中的最高优先级元素（头元素）：

T highestPriorityElement = priorityQueue.peek();

如果队列为空，peek 方法将返回null。

移除并返回队列中的最高优先级元素（头元素）：

T highestPriorityElement = priorityQueue.poll();

如果队列为空，poll 方法将返回null。

检查队列是否为空：

boolean isEmpty = priorityQueue.isEmpty();

获取队列中的元素数量：

int size = priorityQueue.size();

使用自定义比较器：

如果要使用自定义的比较器，可以在创建优先队列时传递一个Comparator 对象，或者使用具有自定义比较器的构造函数：

PriorityQueue customPriorityQueue = new PriorityQueue<>(customComparator);

自定义比较器的示例：

Comparator customComparator = new Comparator() { @Override

public int compare(Integer num1, Integer num2) {

// 返回正数表示num1 的优先级高，返回负数表示num2 的优先级高

return num1 - num2;

}

};

优先队列在许多算法和数据结构中都有广泛应用，例如最小堆、Dijkstra算法、Prim算法等。它允许你高效地管理具有不同优先级的元素，以便在需要时能够快速找到和处理最高优先级的元素。