算法题：剑指 Offer 40. 最小的k个数（题目+思路+代码+注释）时空 O(N*LogN) O(1) 0ms击败56%、32%用户

题目

剑指 Offer 40. 最小的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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

一开始我想直接用快排再取出前面k位，这样有点太赖皮了，于是我试的是我建立一个长度为k的数组，然后从arr里拿前面k个填充进去，并排序，再把剩余的不断的往数组"插入"，该插入操作的步骤是，数组去除最右边的并把这个数插入到它自己的合适位置。后来发现执行用了22ms，然后看了看题解，还是快排最香，最大堆解法也可以。

代码

public int[] getLeastNumbers(int[] arr, int k) {
        //特殊情况直接返回
        if (k == 0 || arr.length == 0){
            return new int[0];
        }
        Arrays.sort(arr);
        int[] nums = new int[k];
        for (int i = 0;i < k;i++){
            nums[i] = arr[i];
        }
        return nums;
    }