最小的K个数

最小的K个数

class Solution {
    /**
     * 使用最大堆找到数组中最小的k个数
     * @param arr 整数数组
     * @param k 需要找的最小数的个数
     * @return 包含最小k个数的数组
     */
    public int[] getLeastNumbers(int[] arr, int k) {
        // 边界情况处理
        if (k == 0 || arr.length == 0) {
            return new int[0];
        }
        
        // 创建最大堆（使用自定义比较器将默认的最小堆转换为最大堆）
        // 比较器 (a, b) -> b - a 表示降序排列，堆顶为最大值
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>((a, b) -> b - a);
        
        // 遍历数组中的所有元素
        for (int num : arr) {
            // 如果堆的大小还不足k，直接加入当前数字
            if (maxHeap.size() < k) {
                maxHeap.offer(num); // 将数字加入堆中
            } 
            // 如果堆已满（已有k个元素），且当前数字比堆顶（当前堆中最大值）小
            else if (num < maxHeap.peek()) {
                maxHeap.poll();    // 移除堆顶的最大值
                maxHeap.offer(num); // 加入当前更小的数字
            }
            // 如果当前数字 >= 堆顶，则跳过（因为我们只需要保留更小的数字）
        }
        
        // 将堆中的元素转换为数组返回
        int[] res = new int[k];
        for (int i = 0; i < k; i++) {
            res[i] = maxHeap.poll(); // 依次取出堆顶元素（从大到小）
        }
        return res;
    }
}