> 文章列表 > PriorityQueue(优先队列)

PriorityQueue(优先队列)

PriorityQueue(优先队列)

在 Java 中,PriorityQueue 是一个队列(Queue)的实现,但它的元素是经过优先级排序的。PriorityQueue 元素的排序可以通过提供所添加元素的比较器或者元素自身实现 Comparable 接口来实现。

PriorityQueue 可以用来实现很多算法,比如 Dijkstra 算法、哈夫曼编码等。

一般来说,PriorityQueue 的时间复杂度是 O(log n),空间复杂度是 O(n)。

下面是使用 PriorityQueue 的一个示例:

import java.util.PriorityQueue;public class PriorityQueueExample {public static void main(String[] args) {// 创建一个整数型的最小优先队列PriorityQueue<Integer> pq = new PriorityQueue<>();// 添加元素pq.offer(3);pq.offer(1);pq.offer(4);pq.offer(2);// 输出队首元素(最小值)System.out.println(pq.peek()); // 输出:1// 删除队首元素(最小值)pq.poll();// 输出队列中元素while (!pq.isEmpty()) {System.out.print(pq.poll() + " "); // 输出:2 3 4}}
}

在上面的代码中,我们首先创建了一个 PriorityQueue,然后使用 offer() 方法添加元素。之后,我们使用 peek() 方法输出队列中的最小值(队首元素),使用 poll() 方法删除队列中的最小值(队首元素),并使用 isEmpty() 方法判断队列是否为空。

需要注意的是,PriorityQueue 默认是从小到大排序,如果我们想要将其变为最大优先队列(即从大到小排序),则可以在创建 PriorityQueue 时传入一个 Comparator 参数:

PriorityQueue<Integer> pq = new PriorityQueue<>(Comparator.reverseOrder());

在实际应用中,我们可以使用 PriorityQueue 来实现一些算法,如 Dijkstra 算法、哈夫曼编码等。