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

Posted 2022-12-11 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;
    

结果