题解 | #最小的K个数#

堆排序方法解题

首先用数组的前k个元素构建一个包含k个元素的大顶堆，存储最小的k个数，再将数组后k个元素跟这个大顶堆的堆顶元素比较，如果小于堆顶元素，说明堆顶元素不是最小的k个数，因此将堆顶元素出堆，将当前元素入堆，堆中会重新排序，重复这样比较，最终堆中就包含了数组中最小的k个数。

参考

https://blog.nowcoder.net/n/51c37cdefe0d4e5d8d0d0120263d9821?f=comment

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param input int整型一维数组 
     * @param k int整型 
     * @return int整型ArrayList
     */
    public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
        // write code here
        ArrayList<Integer> ret = new ArrayList<>();
        if(input.length == 0 || k == 0) {
            return ret;
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b.compareTo(a));  // 优先级队列是个大根堆，维护的是最小的k个数，需要传入一个比较器。
        // 构建一个k大小的堆
        for(int i = 0; i < k; ++i) {
            queue.offer(input[i]);
        }
        for(int i = k; i < input.length; ++i) {
            // 如果数组中的元素小于较小k个数的最大值，说明这个最大值不是最小k个数，所以取出当前最大值，将较小值入队，堆内重新调整排序
            if(queue.peek() > input[i]) {
                queue.poll();
                queue.offer(input[i]);
            }
        }

        ArrayList<Integer> a = new ArrayList<>(queue);
        return a;
    }
}