1.PriorityQueue概述
Java中PriorityQueue通过二叉小顶堆实现,可以用一棵完全二叉树表示(任意一个非叶子节点的权值,都不大于其左右子节点的权值),也就意味着可以通过数组来作为PriorityQueue的底层实现。
父子节点的编号之间有如下关系:
leftNo = parentNo*2+1
rightNo = parentNo*2+2
parentNo = (nodeNo-1)/2
优先队列的作用是能保证每次取出的元素都是队列中权值最小的(Java的优先队列每次取最小元素,及默认是小顶堆)。元素大小的评判可以通过元素本身的自然顺序,也可以通过构造时传入的比较器(如:Comparator)。
2.PriorityQueue构建
大顶堆执行poll()
去掉的是大数,留下的是小数。传入(w1, w2) -> w2 - w1 // 第二个参数减第一个是大顶堆
PriorityQueue<Integer> heap = new PriorityQueue<>(
(w1, w2) -> w2 - w1 ); // 第二个参数减第一个是大顶堆
PriorityQueue默认建立小顶堆执行
poll()`出队的是值小的数据,保留的是值大的数据
PriorityQueue<Integer> heap = new PriorityQueue<>();
PriorityQueue<Integer> queue = new PriorityQueue<Integer>((o1, o2)-> o1 - o2);
3.具体场景
一、找海量数据中最大的K个
构造小根堆,堆顶为最小数,不断入队,当队列数大于K时,将最小的堆顶弹出,最后剩下的就是K个最大数
Leetcode 347. 前 K 个高频元素
题目:给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
【解题思想】
先将k个数放进最小堆,再从第k+1个数开始比较,若其大于小堆顶则加入堆,堆顶出队列,若小于等于则无作为。
class Solution {
public int[] topKFrequent(int[] nums, int k) {
int[] result = new int[k];
HashMap<Integer, Integer> map = new HashMap<>();
for (int num : nums) {
map.put(num, map.getOrDefault(num, 0) + 1);
}
Set<Map.Entry<Integer, Integer>> entries = map.entrySet();
// 根据map的value值正序排,相当于一个小顶堆
PriorityQueue<Map.Entry<Integer, Integer>> queue = new PriorityQueue<>((o1, o2) -> o1.getValue() - o2.getValue());
for (Map.Entry<Integer, Integer> entry : entries) {
queue.offer(entry);
if (queue.size() > k) {
queue.poll();
}
}
for (int i = k - 1; i >= 0; i--) {
result[i] = queue.poll().getKey();
}
return result;
}
}
二、数组中最小的K个
构造最大堆,堆顶为最大数,不断入队,当队列数小于K时,将最大的堆顶弹出,最后剩下的就是K个最小数
剑指offer 40.最小的k个数
输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
示例 1:
输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]
示例 2:
输入:arr = [0,1,2,1], k = 1
输出:[0]
限制:
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000
【解题思想】
先将k个数放进最大堆,再从第k+1个数开始比较,若其小于大堆顶则加入堆,堆顶出队列,若大于等于则无作为。
class Solution {
public int[] getLeastNumbers(int[] arr, int k) {
int[] res = new int[k];
PriorityQueue<Integer> queue = new PriorityQueue<>((o1,o2)-> o2 - o1);//构造大顶堆
for(int num : arr){
queue.offer(num);
if(queue.size() > k){
queue.poll();
}
}
for(int i = 0;i < k;i++){
res[i] = queue.poll();
}
return res;
}
}