29_GetLeastNumbers

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

思路：（该题简单，但找到时间复杂度最小的比较难，冒泡也不是最优）
1）排序
2）选最小的k个
这里可以使用冒泡排序，小的往后冒，只排k轮就行(外层for(i=0;i<k;i++))，不用len-1轮，时间复杂度O(nk)
我使用了插入排序，要把全部元素都排序完，时间复杂度比较高>O(nk)，不如冒泡

package swordToOffer._29_GetLeastNumbers;
import java.util.ArrayList;
import java.util.Arrays;

public class Solution {
    public static void main(String[] args) {
        int[] arr={4,5,1,6,2,7,3,8};
        Solution s = new Solution();
        s.insertSort(arr);
        System.out.println(Arrays.toString(arr));

    }
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if(k>input.length){
            return new ArrayList<>();
        }
        ArrayList<Integer> res = new ArrayList<>();
        insertSort(input);
        for(int i=0;i<k;i++){
            res.add(input[i]);
        }

        return res;
    }
    public void insertSort(int[] arr){
        if(arr.length==0)
            return;
        int lastIndex=0;
        int cur=0;
        for(int i=1;i<arr.length;i++){
            //上一个元素下标
            lastIndex=i-1;
            cur=arr[i];
            //找到待插入的位置lastIndex后一位
            while(lastIndex>=0 && cur<arr[lastIndex]){
                arr[lastIndex+1]=arr[lastIndex];
                lastIndex--;
            }
            if((lastIndex+1)!=i){
                arr[lastIndex+1]=cur;
            }

        }
    }
}