TOPk系列问题

堆排序处理前K系列问题

前k系列问题

找出数组中最小（最大）的k个数。以任意顺序返回这k个数。

一、问题分析

1.若该数组的数组长度很小，则可运用简单方法查找即可。
2.若该数组数组长度很大，超过了内存能存储的范围。那么，显而易见，传统的查找方法已经失去的查找的能力。

二、问题处理

1.堆的建立

a.最大前K个数，建小堆
b.最小前K个数建大堆

示例：建小堆

//向下调整算法
void AdjustDown(int* a,int n,int root)
 {
     int parent=root;
     int child=parent*2+1;
     while(child<n)
     {
          if(child+1<n&&a[child]<a[child+1])
     {
         child=child+1;
     }
     if(a[child]>a[parent])
     {
         int tmp=a[parent];
         a[parent]=a[child];
         a[child]=tmp;

         parent=child;
         child=parent*2+1;
     }
     else
     {
         break;
     }
     }
     

2.数组建堆，进行K堆查找

在堆顶前K个数建立K数大堆，依次比较，若数组arr[k+1]小于K数对顶数retArr[0]，则进行大堆调整，保存小的数，遍历完整个数组，即可找出前K个最小的数，保存在K数堆内。

int* smallestK(int* arr, int arrSize, int k, int* returnSize){
     *returnSize=k;
     int* retArr=(int*)malloc(sizeof(int)*k);

     if(k==0)
        return retArr;
     
     for(int i=0;i<k;i++)
     {
         retArr[i]=arr[i];
     }

     //建K个数的大堆
     for(int i=(k-1-1)/2;i>=0;--i)
     {
        AdjustDown(retArr,k,i);
     }

     for(int j=k;j<arrSize;++j)
     {
         if(arr[j]<retArr[0])
         {
            retArr[0]=arr[j];
            AdjustDown(retArr,k,0);
         }
         
     }
     return retArr;
}

总结

建立K数堆来对数组进行查找，时间复杂度为O（N*logN），解决了因为数组N很大，无法内排序的问题。