1. 程式人生 > >找出一堆數中最小的前K個數

找出一堆數中最小的前K個數

string nlog 浪費 art args 技術分享 str rate .net

描寫敘述:

給定一個整數數組。讓你從該數組中找出最小的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 stub
		int arr[]={9,7,5,4,2,1,3,6,8};
		System.out.println(FirstKElements.getFirstKElements(arr, 4));
	}

}
結果:

技術分享

找出一堆數中最小的前K個數