40.最小的K个数

题目描述(剑指Offer40)

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

示例1
输入

[4,5,1,6,2,7,3,8],4

返回值

[1,2,3,4]

public class Jzof40 {

    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> arrayList = new ArrayList<>();
        if (k > input.length) return arrayList;
        Arrays.sort(input);
        for (int i = 0; i < k; i++) {
            arrayList.add(input[i]);
        }
        return arrayList;
    }
    /**********使用堆************/
    /**********别人的解法********/
    public ArrayList<Integer> GetLeastNumbers_Solution1(int [] input, int k) {
        if (k > input.length || k < 0){
            return new ArrayList<>();
        }
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(((o1, o2) -> o2 - o1));
        for (int i : input) {
            maxHeap.add(i);
            if (maxHeap.size() > k){
                //获取并删除队首元素
                maxHeap.poll();
            }
        }
       return new ArrayList<>(maxHeap);
    }
}

Java 的 PriorityQueue 实现了堆的能力，PriorityQueue 默认是小顶堆，可以在在初始化时使用 Lambda 表达式 (o1, o2) -> o2 - o1 来实现大顶堆