剑指Offer:29.最小的K个数

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

import java.util.*;
//思路：构造一个最小堆构成的最大堆
public class Solution {
    ArrayList<Integer> ans=new ArrayList<>();
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if(k>input.length||k<=0)return ans;
        //构造大顶堆
        PriorityQueue<Integer>pq=new PriorityQueue<>((o1,o2)->(o2-o1));
        for(int num:input){
            pq.offer(num);
            if(pq.size()>k)
                pq.poll();
        }
        return new ArrayList<>(pq);
    }
}