[剑指Offer]-求序列中最小的k个数

问题描述：

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

解法一：

利用快速排序，快速排序每轮排序 可以找出基准的位置，基准前面的序列都小于基准，右边的都大于基准。利用这个特性求解。但是这种解法会改变数组的结构。暂时先不给出这种实现的源码。

解法二：适用于大数据下

比如在5000千万个数据下，找出这5000千万个数据中最下的10个数据。

解题思路：因为数据量的巨大，不可能进行排序，占用内存会巨大。所以这里利用一个容器，先往容器里面存入10个数据，然后找出这个容器的最大值，然后一次将5000千万数据中剩余的数据一个个取出来与容器中最大值对比，如果大于容器最大值，直接过滤掉，如果小于容器最大值，就删除容器的最大值，将这个新值插入容器中。保证容器中数据一直是最小的10个数据。遍历完5000千万个数据就选出最小的10个数据。

这里最大的问题是怎么实现这个容器，保证容器中数据一直是有序的，能够直接选出最大值。最简单的是实现是最大堆，堆顶一直是最大值，但是实现一个最大堆花费时间多，代码量也大。这里我用Java的TreeMap来实现，TreeMap是基于红黑树的，默认情况下Key是有序的，自动维护key的排序性。所以可以满足容器的需求。

这里我写了check() 函数，判断输入参数temp 校验TreeMap的容器。实现了容器的逻辑需求。

public static void check(int temp, TreeMap<Integer, String> treeMap){
    int max = treeMap.lastKey();//获取键的最大值

    if(max > temp){
        //删除max 插入temp
        treeMap.remove(max);
        treeMap.put(temp, String.valueOf(temp));
    }else{
        return;
    }
}

给出了上面的功能之后，再实现题目的要求就很简单了。