每日一题-2.最小的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]

限制:

0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

思考一:

寻找k个最小数,首先建立一个k大小的数组，将原数组的前k个数依次放入到该数组中，继续循环当前原数组，并且每次找到k大小数组的最大值和当前元素比较，如果发现k数组的最大值大于当前元素，则将k数组元素的最大值替换为当前元素，遍历完原数组，则k数组存放的元素则是最小的k个数。
代码:

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
         int result[] = new int[k];
         if(k==0||arr.length==0){
            return new int[0];
         }
         for(int i=0;i < arr.length;i++){
             if(i<k){
                result[i]=arr[i];
             }else{
                 int index = findMaxIndex(result);
                 if(result[index]>arr[i]){
                     result[index]=arr[i];
                 }
             }
         }
         return result;
    }

    public static int findMaxIndex(int arr[]){
         if(arr.length==1){
            return 0;
         }
         int max = 0;
         for(int i=1;i<arr.length;i++){
            if(arr[i]>arr[max]){
                max = i;
            }
         }
         return max;
    }
}

运行实例:

执行用时 :174 ms, 在所有 Java 提交中击败了8.09%的用户
内存消耗 :42.6 MB, 在所有 Java 提交中击败了100.00%的用户

根据提交可以看出，运行效率不是那么高，考虑下有没有可以优化的地方。

思考二:

public int[] getLeastNumbers(int[] arr, int k) {
    if (k == 0) {
        return new int[0];
    }
    // 使用一个最大堆（大顶堆）
    // Java 的 PriorityQueue 默认是小顶堆，添加 comparator 参数使其变成最大堆
    Queue<Integer> heap = new PriorityQueue<>(k, (i1, i2) -> Integer.compare(i2, i1));

    for (int e : arr) {
        // 当前数字小于堆顶元素才会入堆
        if (heap.isEmpty() || heap.size() < k || e < heap.peek()) {
            heap.offer(e);
        }
        if (heap.size() > k) {
            heap.poll(); // 删除堆顶最大元素
        }
    }

    // 将堆中的元素存入数组
    int[] res = new int[heap.size()];
    int j = 0;
    for (int e : heap) {
        res[j++] = e;
    }
    return res;
}