题解 | #最小的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> res = new ArrayList<Integer>();
        //建立大根堆，建立优先队列使用lambda表达式 初始化方式 为后者-前者
        PriorityQueue<Integer> p = new PriorityQueue<Integer>((p1, p2)
                ->p2 - p1);
        if(k==0||input.length==0){
            return res;
        }
        //先存满k个元素
        for (int i = 0; i < input.length; i++) {
            if (p.size() < k) {
                p.offer(input[i]);
            } else {
                if (input[i] < p.peek()) {
                    p.poll();
                    p.offer(input[i]);
                }
            }
        }
        for(int num:p){
            res.add(num);
        }
        // Iterator<Integer> iterator = p.iterator();
        // while (iterator.hasNext()) {
        //     res.add(iterator.next());
        // }
       Collections.sort(res);
        return res;
    }
}