【數(shù)據(jù)結(jié)構(gòu)】找出N個(gè)數(shù)據(jù)中最大的前k個(gè)數(shù)據(jù)（利用堆排序）

我們舉例，假若從10000萬(wàn)個(gè)數(shù)里選出前100個(gè)最大的數(shù)據(jù)。

創(chuàng)新互聯(lián)建站-專業(yè)網(wǎng)站定制、快速模板網(wǎng)站建設(shè)、高性價(jià)比貴定網(wǎng)站開(kāi)發(fā)、企業(yè)建站全套包干低至880元,成熟完善的模板庫(kù),直接使用。一站式貴定網(wǎng)站制作公司更省心,省錢(qián),快速模板網(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ù)必然在它頂端。假若真不巧，我開(kāi)始的前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
#include<iostream>
using namespace std;

#include<assert.h>

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]<a[i])
        {
            arr[0] = a[i];
            AdjustDown(arr, 0, K);
        }
    }

    return arr;
    delete[] arr;
}


void Test()
{
    int arr[] = { 12, 2, 10, 4, 6, 8, 54, 67, 25, 178 };
    int k = 5;
    int* ret = HeapSet(arr, sizeof(arr) / sizeof(arr[0]), k);
    Print(ret, k); 
}


int main()
{
    Test();
    system("pause");
    return 0;
}

由此可以看出，時(shí)間復(fù)雜度為：Ｋ＋（Ｋ-2）/２*lgn＋（Ｎ－Ｋ）*lgn --＞ O(N)

空間復(fù)雜度為：K-->O(1)。

名稱欄目：【數(shù)據(jù)結(jié)構(gòu)】找出N個(gè)數(shù)據(jù)中最大的前k個(gè)數(shù)據(jù)（利用堆排序）
網(wǎng)頁(yè)地址：http://muchs.cn/article48/ghgghp.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供虛擬主機(jī)、企業(yè)建站、App開(kāi)發(fā)、、企業(yè)網(wǎng)站制作、標(biāo)簽優(yōu)化

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源：創(chuàng)新互聯(lián)

猜你還喜歡下面的內(nèi)容