我們舉例,假若從10000萬(wàn)個(gè)數(shù)里選出前100個(gè)最大的數(shù)據(jù)。
創(chuàng)新互聯(lián)建站-專業(yè)網(wǎng)站定制、快速模板網(wǎng)站建設(shè)、高性價(jià)比貴定網(wǎng)站開發(fā)、企業(yè)建站全套包干低至880元,成熟完善的模板庫(kù),直接使用。一站式貴定網(wǎng)站制作公司更省心,省錢,快速模板網(wǎng)站建設(shè)找我們,業(yè)務(wù)覆蓋貴定地區(qū)。費(fèi)用合理售后完善,十余年實(shí)體公司更值得信賴。
首先我們先分析:既然要選出前100個(gè)最大的數(shù)據(jù),我們就建立一個(gè)大小為100的堆(建堆時(shí)就按找最大堆的規(guī)則建立,即每一個(gè)根節(jié)點(diǎn)都大于它的子女節(jié)點(diǎn)),然后再將后面的剩余數(shù)據(jù)若符合要求就插入堆中,不符合就直接丟棄該數(shù)據(jù)。
那我們現(xiàn)在考慮:確定是該選擇最大堆的數(shù)據(jù)結(jié)構(gòu)還是最小堆的數(shù)據(jù)結(jié)構(gòu)呢。
分析一下:
若選用最大堆的話,堆頂是堆的最大值,我們考慮既然要選出從10000萬(wàn)個(gè)數(shù)里選出前100個(gè)最大的數(shù)據(jù),我們?cè)诮ǘ训臅r(shí)候,已經(jīng)考慮了最大堆的特性,那這樣的話最大的數(shù)據(jù)必然在它頂端。假若真不巧,我開始的前100個(gè)數(shù)據(jù)中已經(jīng)有這10000個(gè)數(shù)據(jù)中的最大值了,那對(duì)于我后面剩余的10000-100的元素再想入堆是不是入不進(jìn)去了?。?!所以,選用最大堆從10000萬(wàn)個(gè)數(shù)里選出前100個(gè)最大的數(shù)據(jù)只能找出一個(gè),而不是100個(gè)。
那如果選用最小堆的數(shù)據(jù)結(jié)構(gòu)來(lái)解決,最頂端是最小值,再次遇到比它大的值,就可以入堆,入堆后重新調(diào)整堆,將小的值pass掉。這樣我們就可以選出最大的前K個(gè)數(shù)據(jù)了。言外之意,假若我們要找出N個(gè)數(shù)據(jù)中最小的前k個(gè)數(shù)據(jù),就要用最大堆了。
代碼實(shí)現(xiàn)(對(duì)于最大堆最小堆的代碼,若有不明白的地方,大家可以查看我的博客http://10740184.blog.51cto.com/10730184/1767076):
#define _CRT_SECURE_NO_WARNINGS 1 #includeusing namespace std; #include void AdjustDown(int* a, int parent, int size) { int child = 2 * parent + 1; while (child < size) { if (child + 1 < size && a[child] > a[child + 1]) { child++; } if (a[parent]>a[child]) { swap(a[parent], a[child]); parent = child; child = 2 * parent + 1; } else { break; } } } void Print(int* a, int size) { cout << "前k個(gè)最大的數(shù)據(jù):" << endl; for (int i = 0; i < size; i++) { cout << a[i] << " "; } cout << endl; } int* HeapSet(int*a,int N,int K) { assert(a); assert(K > 0); int* arr = new int[K]; //將前K個(gè)數(shù)據(jù)保存 for (int i = 0; i < K; i++) { arr[i] = a[i]; } //建堆 for (int i = (K-2)/2; i >=0; i--) { AdjustDown(arr,i,K); } //對(duì)剩余的N-K個(gè)元素比較大小 for (int i = K; i < N; i++) { if (arr[0] 由此可以看出,時(shí)間復(fù)雜度為:K+(K-2)/2*lgn+(N-K)*lgn --> O(N)
空間復(fù)雜度為:K-->O(1)。
網(wǎng)站標(biāo)題:【數(shù)據(jù)結(jié)構(gòu)】找出N個(gè)數(shù)據(jù)中最大的前k個(gè)數(shù)據(jù)(利用堆排序)
標(biāo)題鏈接:http://weahome.cn/article/ghgghp.html