优先级队列面试题详解

题目链接：

. - 力扣（LeetCode）. - 备战技术面试？力扣提供海量技术面试资源，帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/smallest-k-lcci/description/

---

 

解题思路：实际上，这题属于优先级队列中的Top-K问题，需要注意的是我们建立的优先级队列默认是小根堆，本题要找的是最小的k个数，应实现Comparator接口，重写compare方法。

---

 

代码实现：

class Cmp implements Comparator<Integer>{
//想要改为大根堆，实现Comparator接口，重写compare方法
    public int compare(Integer o1, Integer o2) {
        return o2.compareTo(o1);
    }
}

class Solution {
    public int[] smallestK(int[] arr, int k) {
        if(k==0){
            return new int[k];
        }

        
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(new Cmp());
        // 将数组前k个元素放入优先级队列中，建立大根堆
        for (int i = 0; i < k; i++) {
            priorityQueue.offer(arr[i]);
        }

        int j = k ;
        int[] array = new int[k];

        //结束条件--数组遍历完成
        while (j < arr.length) {
            if (arr[j] < priorityQueue.peek()) {
                priorityQueue.poll();
                priorityQueue.offer(arr[j]);
            }
            j++;
        }
        
        //筛选好后放入数组中用于放回
        for (int z = 0; z < k; z++) {
            array[z] = priorityQueue.poll();
        }

        return array;
    }

}