题解 | #最小的K个数#

import java.util.*;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        if(input.length < k || k < 1 || input.length <0) return list;


        //创建一个大根堆 堆顶为堆中的最大元素 ，
        //每次取出堆顶元素 就相当于取出堆中的最大元素
        //注意 堆中元素是无序存储的 
        Queue<Integer> queue = new PriorityQueue<>(k,Collections.reverseOrder()); 

        for(int i = 0 ; i < input.length ; i++){
                if(queue.size() < k){
                    queue.add(input[i]);
                }else{
                    if(input[i] < queue.peek()){
                        //如果 当前元素小于 堆中最大元素 
                        //将堆中最大元素弹出 将当前元素加入堆中
                        queue.poll();
                        queue.add(input[i]);
                    }
                }
        }
        while(!queue.isEmpty()){
            list.add(queue.poll());
        }



        return list;
    }
}