题解 | #最小的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
        // 结果集
        ArrayList<Integer> result = new ArrayList<>();
        if (input == null || input.length == 0 || k == 0 || k > input.length) {
            return result;
        }
        // 使用小顶堆
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
        for (int i = 0; i < input.length; i ++) {
            priorityQueue.offer(input[i]);
        }
        int j = 1;
        while (j <= k && !priorityQueue.isEmpty()) {
            result.add(priorityQueue.poll());
            j++;
        }
        return result;
    }
}