在new 堆的时候,错了很多遍。
构建大顶堆需要重写Comparator的compare方法。
Java集合如Map、Set、List等所有集合只能存放引用类型数据,它们都是存放引用类型数据的容器,不能存放如int、long、float、double等基础类型的数据。
所以PriorityQueue、Comparator的类型以及compare的参数都不能写成int
PriorityQueue<Integer> heap = new PriorityQueue<>(
new Comparator<Integer>(){
public int compare(Integer num1, Integer num2){
return num2 - num1;
}
}
);
class Solution {
public int[] smallestK(int[] arr, int k) {
PriorityQueue<Integer> heap = new PriorityQueue<>(
new Comparator<Integer>(){
public int compare(Integer num1, Integer num2){
return num2 - num1;
}
}
);
for(int i = 0; i < arr.length; i++){
heap.add(arr[i]);
if(heap.size() > k ){
heap.poll();
}
}
int [] res = new int [k];
for(int i = 0; i < k; i++){
res[i] = heap.poll();
}
return res;
}
}