题解 | #最小的K个数#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param input int整型一维数组
     * @param k int整型
     * @return int整型ArrayList
     */
    public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        if (k == 0) {
            return list;
        }
        // 建堆（大顶堆）
        PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);
        for (int i = 0; i < k; i++) {
            queue.offer(input[i]);
        }

        // 将数组剩余元素和堆顶元素比较，比堆顶元素小，则将堆顶元素弹出，将该元素加入堆
        for (int i = k; i < input.length; i++) {
            if (queue.peek() > input[i]) {
                queue.poll();
                queue.offer(input[i]);
            }
        }

        // 最后取出堆里的元素即可
        for (int i = 0; i < k; i++) {
            list.add(queue.poll());
        }
        return list;
    }
}