找出一堆数中最小的前K个数

                                      找出一堆数中最小的前K个数

描述：

给定一个整数数组，让你从该数组中找出最小的K个数

思路：

最简洁粗暴的方法就是将该数组进行排序，然后取最前面的K个数即可。但是，本题要求的只是求出最小的k个数即可，用排序可以但显然有点浪费，比如让求10000个整数数组中的最小的10个数，用排序的话平均时间复杂度差为Nlog（N）。

于是想到了，用堆来实现，但是自己实现又太麻烦，想到了Java里面的TreeSet，先将K个数放入TreeSet中，由于TreeSet会对里面的元素进行排序，所以在TreeSet中的元素是有序的，以后没插入一个元素，将TreeSet中的最大元素删除即可，所以TreeSet动态维持K个元素且这K个元素是有序的。

代码： copy

1. import java.util.TreeSet;  

2.   

3. public class FirstKElements  

4. {  

5.     public static TreeSet<Integer>getFirstKElements(int arr[],int k)  

6.     {  

7.         TreeSet<Integer>set=new TreeSet<Integer>();  

8.         int len=arr.length;  

9.         k=k%len;  

10.         int i=0;  

11.         int num=0;  

12.         for(i=0;i<k;i++)  

13.             set.add(arr[i]);  

14.         for(i=k;i<len;i++)  

15.         {  

16.             set.add(arr[i]);  

17.             num=set.last();  

18.             set.remove(num);  

19.         }  

20.         return set;  

21.     }  

22.     public static void main(String[] args)  

23.     {  

24.         // TODO Auto-generated method stub  

25.         int arr[]={9,7,5,4,2,1,3,6,8};  

26.         System.out.println(FirstKElements.getFirstKElements(arr, 4));  

27.     }  

28.   

29. }  

结果：