题目链接:
. - 力扣(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;
}
}