题解 | #最小的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，使用优先队列，生成顶最小队列
        PriorityQueue<Integer> q = new PriorityQueue<>();
        for(int i = 0; i < input.length; i ++){
            q.offer(input[i]);
        }
        if(q.size() < k) return null;
        ArrayList<Integer> res = new ArrayList<>();
        for(int i = 0; i < k; i ++){
            res.add(q.poll());
        }
        return res;
    }
}