描寫敘述：

給定一個整數數組。讓你從該數組中找出最小的K個數

思路：

最簡潔粗暴的方法就是將該數組進行排序，然后取最前面的K個數就可以。

可是，本題要求的僅僅是求出最小的k個數就可以，用排序能夠但顯然有點浪費。比方讓求10000個整數數組中的最小的10個數。用排序的話平均時間復雜度差為Nlog（N）。

于是想到了，用堆來實現，可是自己實現又太麻煩。想到了java里面的TreeSet，先將K個數放入TreeSet中。因為TreeSet會對里面的元素進行排序。所以在TreeSet中的元素是有序的。以后沒插入一個元素，將TreeSet中的最大元素刪除就可以，所以TreeSet動態維持K個元素且這K個元素是有序的。

代碼：

import java.util.TreeSet;public class FirstKElements {public static TreeSet<Integer>getFirstKElements(int arr[],int k){TreeSet<Integer>set=new TreeSet<Integer>();int len=arr.length;k=k%len;int i=0;int num=0;for(i=0;i<k;i++)set.add(arr[i]);for(i=k;i<len;i++){set.add(arr[i]);num=set.last();set.remove(num);}return set;}public static void main(String[] args){// TODO Auto-generated method stubint arr[]={9,7,5,4,2,1,3,6,8};System.out.println(FirstKElements.getFirstKElements(arr, 4));}}結果：

總結

以上是生活随笔為你收集整理的找出一堆数中最小的前K个数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。