找到数组中最小的k个数

    /*输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，
    则最小的4个数字是1、2、3、4。
    示例 1：
    输入：arr = [3,2,1], k = 2
    输出：[1,2] 或者 [2,1]
    示例 2：
    输入：arr = [0,1,2,1], k = 1
    输出：[0]
    https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/*/
    //最小的k个数
    public static int[] getLeastNumbers(int[] arr, int k) {
       /* PriorityQueue<Integer> pQueue = new PriorityQueue<Integer>(arr.length, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o1 - o2;
            }
        });*/
        PriorityQueue<Integer> pQueue = new PriorityQueue<Integer>();//默认小顶堆
        for (int i:arr
             ) {
            pQueue.add(i);
        }
        int[] result = new int[k];
        for (int i = 0; i < k; i++) {
            result[i] = pQueue.poll();
        }
        return result;
    }

//最小的k个数
public static int[] getLeastNumbers2(int[] arr, int k) {
    //大顶堆
    PriorityQueue<Integer> pQueue = new PriorityQueue<Integer>(new Comparator<Integer>() {
        @Override
        public int compare(Integer o1, Integer o2) {
            return o2-o1;
        }
    });//默认小的在上
    for (int i:arr
    ) {
        if(pQueue.size() < k ){
            pQueue.add(i);
        }else if(!pQueue.isEmpty()&&i<pQueue.peek()){
            pQueue.poll();
            pQueue.add(i);
        }
    }
    int[] result = new int[k];
    for (int i = 0;i<k ;i++) {
        result[i] = pQueue.poll();
    }
    return result;
}