剑指offer--40最小的k个数

Posted 2021-08-27 Anrys

剑指offer--40最小的k个数

• 题目
• 代码
• 结果

题目

代码

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if (k == 0 || arr.length == 0) return new int[0];
        //实现升序排列即头节点是最大的--需要重写一下比较器。
        Queue<Integer> pq = new PriorityQueue<>((v1, v2) -> v2-v1);
        for (int num: arr) {
            if (pq.size() < k) pq.offer(num);
            else if (num < pq.peek()) {
                pq.poll();
                pq.offer(num);
            }
        }
        // 返回堆中的元素
        int[] res = new int[pq.size()];
        int idx = 0;
        for(int num: pq) { res[idx++] = num; }
        return res;
    }
}

结果