堆的应用之TOP K问题

此问题需要在N个数中找出最大或者最小的K 个数。这里我们用找最大的K个数来举例。

void AdjustDown(int* hp, int K, int i)  
{  
    int parent = i;  
    int child = i * 2 + 1;  
    while (child < K)  
    {  
        if (child + 1 < K&&hp[child + 1] < hp[child])  
        {  
            ++child;  
        }  
        if (hp[parent]>hp[child])  
        {  
            swap(hp[child], hp[parent]);  
            parent = child;  
            child = parent * 2 + 1;  
        }  
        else  
        {  
            break;  
        }  
    }  
}  


///TopK问题  
void TopK()  
{  
    const int N = 1000;  // 假设给随机1000个数  
    const int K = 5;     //找最大的5个数  
    int a[N] = {0};        
    for (int i = 0; i < N; ++i)  
    {  
        a[i] = rand() % 1000;  //创建数组a[]  
    }  
    a[2] = 5000;      //给定几个最大的数，以便测试代码时，这几个数一定在其中  
    a[62] = 1020;  
    a[996] = 8888;  
    a[452] = 9999;  

    int hp[K] = { 0 };  
    for (int i = 0; i < K; ++i)   //用a[]数组中的K个数创建hp[]这个数组  
    {  
        hp[i] = a[i];  
    }  

    for (int i = (K - 2) >> 2; i >= 0; --i) 用数组hp建立小堆  
    {  
        AdjustDown(hp, K, i);  
    }  
    for (size_t i = K; i < N; ++i)     拿建好的小堆与a[]数比较。（小堆时，hp[0]是堆中最小的数，只要a[i]比它大就交换）  
    {  
        if (a[i]>hp[0])  
        {  
            hp[0] = a[i];  
            AdjustDown(hp, K, 0);  
        }  
    }  

    for (size_t i = 0; i < K; ++i)   打印这个堆，可以看到最大的五个数  
    {  
        cout << hp[i] << " ";  
    }  
    cout << endl;  
}