PriorityQueue优先级队列

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;
    }
}