利用K-堆求最大的K个数

1.原理：

1.1将数据存在数组中，利用堆排排序数组的前K个数（递增），此时建立的是小堆，数组的第一个元素保存堆中最小值。

1.2从（K+1）个数组元素遍历到最后，每次都和堆顶元素对比，如果大于堆顶元素，交换两者，调整堆。

1.3输出数组前K个数，就是最大的K个数。

#include<stdio.h>
#define K 3

void swap(int *a,int *b)
{
    int *temp=NULL;
    *temp=*a;
    *a=*b;
    *b=*temp;
}
void shift(int arr[],int n,int i)
{
    int j,temp;
    j=2*i+1;
    temp=arr[i];
    while(j<n)
    {
        if(j+1<n&&arr[j]>arr[j+1])
            ++j;
        if(arr[j]>temp)
            break;
        arr[i]=arr[j];
        i=j;
        j=2*i+1;
    }
    arr[i]=temp;
}
void createK(int arr[],int n)
{
     int i;
     shift(arr,K-1,0);
     shift(arr,K,0);
     printf("min=%5d\n",arr[0]);
     for(i=K;i<n;++i)
      {
            if(arr[i]>arr[0])
                {
                    swap(&arr[i],&arr[0]);
                    shift(arr,K,0);
                }
        }
}

int main()
{
    int i;
    int arr[6]={77,6,7,12,5,9};
    createK(arr,6);
    for(i=0;i<K;++i)
        printf("%5d",arr[i]);
    getch();
    return 0;
}

疑问：K-堆就是对前面的K 个数排序。那么利用任何一种改进的排序方法排序前K个数，时间复杂度都是O(N*logk).