最小的k个数

1.题目

2.解法（使用插入排序）

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        
        ArrayList<Integer> result = new ArrayList<Integer>();
        if(k<= 0 || k > input.length)return result;
        //初次排序，完成k个元素的排序
         for (int i = 1; i < k; i++) {
             int j = i;
             int tmp = input[i];
             while(j > 0 && input[j-1] > tmp) {
                 input[j] = input[j-1];
                 --j;
             }
             input[j] = tmp;
         }
        
        //遍历后面的元素 进行k个元素的更新和替换
        for (int i = k; i < input.length; i++) {
            if (input[i] < input[k-1]) {
                int tmp = input[i];
                int j = k - 1;
                while (j > 0 && input[j-1] > tmp) {
                    input[j] = input[j-1];
                    --j;
                }
                input[j] = tmp;
            }
        }
        

        //把前k个元素返回
        for (int i = 0; i < k; i++)
            result.add(input[i]);
        return result;
    }
}

先排好前k个数，然后再1个1个的比
== 最坏情况下时间复杂度为O(n^2)，空间复杂度为O(k)==