题解 | 最小的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) {
        // write code here
        //维护一个最大堆
        if (k == 0) return new ArrayList<>(); // 边界条件
        PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) -> o2.compareTo(o1));

        for (int it: input) {
            if (pq.size() < k) {
                pq.offer(it);
            } else {
                if (pq.peek() > it) {
                    pq.poll();
                    pq.offer(it);
                }
            }
        }  

        ArrayList<Integer> list = new ArrayList<>();
        while (!pq.isEmpty() && k > 0) {
            list.add(0, pq.poll());
            k--;
        }
        return list;
        


    }
}