最小的K个数

题目描述

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

思路一：
先将整个数组排序，之后取其中的前k个。

public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
    ArrayList<Integer> res = new ArrayList<Integer>();
    if(input.length == 0) return res;
    if(input.length < k) return res;
    //sort的API内部用快排实现，时间复杂度O(nlog(n)) 
    Arrays.sort(input);
    for(int i=0;i<k;i++){
        res.add(input[i]);
    }
    return res;
}

思路二：
先去头k个数存到ArrayList中，之后在从第k+1个开始变了数组，替换ArrayList中最大的数字，最后返回的就是k个最小的。

public ArrayList<Integer> GetLeastNumbers_Solution1(int[] input, int k) {
    ArrayList<Integer> list = new ArrayList<Integer>();

    if (input.length < k || k == 0)
        return list;

    for (int i = 0; i < k; i++)
        list.add(input[i]);

    for (int i = k; i < input.length; i++) {
        int j = this.getMax(list);
        int temp = (Integer) list.get(j);
        if (input[i] < temp)
            list.set(j, input[i]);
    }
    return list;
}

public int getMax(ArrayList<Integer> list) {
    int max = list.get(0);
    int j = 0;
    for (int i = 0; i < list.size(); i++) {
        if (list.get(i) > max) {
            max = list.get(i);
            j = i;
        }
    }
    return j;
}