JZ29 最小的K个数

保持堆的大小为K，然后遍历数组中的数字，遍历的时候做如下判断：

1. 若目前堆的大小小于K，将当前数字放入堆中；
2. 否则判断当前数字与大根堆堆顶元素的大小关系，如果当前数字比大根堆堆顶还大，这个数就直接跳过；
3. 反之如果当前数字比大根堆堆顶小，先poll掉堆顶，再将该数字放入堆中。

import java.util.*;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res = new ArrayList<>();
        if(input != null && k > 0 && input.length >= k){
            PriorityQueue<Integer> pq = new PriorityQueue<>(k, (v1, v2) -> (v2 - v1));
            for(int num: input){
                if(pq.size() < k){
                    pq.offer(num);
                }else if(num < pq.peek()){
                    pq.poll();
                    pq.offer(num);
                }
            }
            while(!pq.isEmpty()){
                res.add(pq.poll());
            }
        }
        return res;
    }
}