找出一堆数中最小的前K个数
生活随笔
收集整理的這篇文章主要介紹了
找出一堆数中最小的前K个数
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
描寫(xiě)敘述:
給定一個(gè)整數(shù)數(shù)組。讓你從該數(shù)組中找出最小的K個(gè)數(shù)
思路:
最簡(jiǎn)潔粗暴的方法就是將該數(shù)組進(jìn)行排序,然后取最前面的K個(gè)數(shù)就可以。
可是,本題要求的僅僅是求出最小的k個(gè)數(shù)就可以,用排序能夠但顯然有點(diǎn)浪費(fèi)。比方讓求10000個(gè)整數(shù)數(shù)組中的最小的10個(gè)數(shù)。用排序的話平均時(shí)間復(fù)雜度差為Nlog(N)。
于是想到了,用堆來(lái)實(shí)現(xiàn),可是自己實(shí)現(xiàn)又太麻煩。想到了java里面的TreeSet,先將K個(gè)數(shù)放入TreeSet中。因?yàn)門(mén)reeSet會(huì)對(duì)里面的元素進(jìn)行排序。所以在TreeSet中的元素是有序的。以后沒(méi)插入一個(gè)元素,將TreeSet中的最大元素刪除就可以,所以TreeSet動(dòng)態(tài)維持K個(gè)元素且這K個(gè)元素是有序的。
代碼:
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));}}結(jié)果:總結(jié)
以上是生活随笔為你收集整理的找出一堆数中最小的前K个数的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: Fastjson 爆出远程代码执行高危漏
- 下一篇: LinksysWrt1900acs刷官方