题解 | #最小的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(input.length == 0 || k<= 0 || k > input.length){
            return new ArrayList<>();
        }
        /*
        建立一个优先队列，默认是最小堆，加添Comparator.reverseOrder()参数之后为最大堆
        */
        PriorityQueue<Integer> queue = new PriorityQueue<>(Comparator.reverseOrder());

        for(int num: input){
            if(queue.size() < k){
                queue.offer(num);
            }else if(num < queue.peek()){
                queue.poll();
                queue.offer(num);
            }
        }
        ArrayList<Integer> list = new ArrayList<>();
        while(!queue.isEmpty()){
            list.add(queue.poll());
        }
        //反转arraylist
        Collections.reverse(list);
        return list;
    }
}