题解 | 最小的K个数

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param input int整型一维数组 
     * @param k int整型 
     * @return int整型ArrayList
     */
    // 最优方法是使用优先队列（堆排序）
    // PriorityQueue，是一种内置的机遇堆排序的容器，分为大顶堆与小顶堆，大顶堆的堆顶为最大元素，其余更小的元素在堆下方，小顶堆与其刚好相反。且因为容器内部的次序基于堆排序，因此每次插入元素时间复杂度都是O(log2n)，而每次取出堆顶元素都是直接取出
    public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
        // write code here
        ArrayList<Integer> res = new ArrayList<>();
        if(k == 0 || input.length == 0) return res;
        // 大根堆
        PriorityQueue<Integer> q = new PriorityQueue<>((o1,o2) -> o2.compareTo(o1));
        // 构建一个k个大小的堆
        for(int i = 0;i<k;i++){
            q.offer(input[i]);
        }
        for(int i = k;i<input.length;i++){
            // 较小元素入堆
            if(q.peek() > input[i]){
                q.poll();
                q.offer(input[i]);
            }
        }
        // 堆中元素取出入数组
        for(int i = 0;i<k;i++){
            res.add(q.poll());
        }
        return res;
    }
}