题解 | 最小的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>();
        if(k==0||input.length==0)
            return res;
        PriorityQueue<Integer> q=new PriorityQueue<>((o1,o2)->o2.compareTo(o1));
        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;
    }
}

以后凡是这种求最大的K个数，最小的K个数的题目，记得考虑堆排序。

在java里面堆排序一般用有限队列来实现，然后有限队列默认是小顶堆，所以需要自己构建一个比较器，进而构建大顶堆。这里面的方法首先构建了一个大小为k的大顶堆。然后将剩下的元素与堆顶元素比较，如果比堆顶元素小，就将堆顶元素抛弃，然后将新元素入堆。最后将堆里面的所有元素返回即可。