堆中等 leetcode面试题 17.14. 最小K个数

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

堆中等 leetcode面试题 17.14. 最小K个数_第1张图片

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

你可能感兴趣的:(leetcode做题笔记,#,堆)