剑指offer系列之28：最小的k个数

题目描述：输入n个数，找出其中最小的k个数。

思路1：最简单的思路就是把输入的n个整数排序，排序之后位于最前面的k个数就是最小的k个数。其实可以使用java的api来完成排序。

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(k>input.length){
            return list;
        }
        int temp;
        for(int i = input.length - 1;i>=0;i--){
            for(int j = 0;j<i;j++){
                if(input[j]>input[j+1]){
                    temp = input[j+1];
                    input[j+1] = input[j];
                    input[j] = temp;
                }
            }
        }


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

        return list;
    }
}

思路2：创建一个用于保存最小k个数的容器，第一步是放入k个元素，第二步是比较容器中最大的数与当前数组遍历的那个数的大小，如果容器中最大的数比当前遍历的数大，就移除该数，并放入当前遍历的那个数。这样整个循环结束的时候，容器中就是最小的k个数了。就是一致维护一个有k个数的大根堆，这个堆代表目前选出的k个最小的数，在堆里的k个元素中堆顶元素是最小的k个元素里面最大的那个。可以参考左老师书上的代码。

public ArrayList<Integer> GetLeastNumbers_Solution2(int[] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(input == null || k <= 0 || input.length == 0 || k > input.length) return list;
        //创建一TreeSet集合，因为TreeSet默认就是从小到大排序的
        TreeSet<Integer> s = new TreeSet<Integer>();
        for (int i = 0; i < input.length; i++) {
            if(s.size() < k){
                s.add(input[i]);
            }else{
                if(input[i] < s.last()){
                    s.remove(s.last());
                    s.add(input[i]);
                }
            }
        }
        //循环结束，s中已经保存了前k个最小的数，只需要迭代放入ArrayList中即可
        Iterator<Integer> it = s.iterator();
        while(it.hasNext()){
            list.add(it.next());
        }
        return list;
    }