在 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 算法、哈夫曼编码等。
相似的:比较器
int[][] arr = {{4, 2, 1}, {9, 3, 5}, {8, 7, 6}};
// 按第一列升序排序
Arrays.sort(arr, Comparator.comparingInt(a -> a[0]));
// 按第二列降序排序
Arrays.sort(arr, Comparator.comparingInt((int[] a) -> a[1]).reversed());
// 按第三列升序排序,如果值相同,则按第一列升序排序
Arrays.sort(arr, Comparator.comparingInt((int[] a) -> a[2]).thenComparingInt(a -> a[0]));