TopK排序

利用快速排序实现TopK排序

 //返回支点的下标
 int partition(int *arr, int low, int high)
 {
     //选取第一个元素为支点
     int pivot = arr[low];
     while(low < high)
     {
         //从右往左
         while(low < high && arr[high] >= pivot)
             high--;
         //该元素小于支点，移前去
         arr[low] = arr[high];

         //从左往右
         while(low < high && arr[low] <= pivot)
             low++;
         //该元素大于支点，移到后面
         arr[high] = arr[low];
         //将支点移至最终的位置
         arr[low] = pivot;
     }
     return low;
 }

 //递归找到前K个数
 int selectK(int *arr, int low, int high, int k)
 {
     int index = ;
     if(low < high)
     {
         index = partition(arr,low,high);
         if(k == index)
             index = k;//返回K
         else if(index < k)
             index = selectK(arr,index+,high,k-index);//还要从右边找到k-index个数
         else if(index > k)//从左边找前K个数
             index = selectK(arr,low,index-,k);
     }
     return index;
 }