题解 | #最小的K个数#

//优先队列（大根堆）的使用，它内部会自动调整结构，大的在peek
//ArrayList在指定位置插入元素，后面的元素会自动右移

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> array = new ArrayList<>(k);//指定初始容量
        if(input.length == 0 || k == 0){
            //return null;//啥都没有
            return array;//返回一个空集合[]
        }
        // 使用大根堆（优先队列）
        PriorityQueue<Integer> pr = new PriorityQueue<>((o1,o2)->o2.compareTo(o1));
        for(int i=0;i<k;i++){
            pr.add(input[i]);
        }
        for(int i=k;i<input.length;i++){
            if(input[i]<pr.peek()){
                pr.poll();//弹出
                pr.offer(input[i]);
            }
        }
        //以上操作，得到最小的k个值的大根堆
        for(int i=0;i<k;i++){
            array.add(0,pr.poll());//源码可见，插入后默认向右移位
        }
        return array;
    }
}