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 = 0;
    if(low < high)
    {
        index = partition(arr,low,high);
        if(k == index)
            index = k;//返回K
        else if(index < k)
            index = selectK(arr,index+1,high,k-index);//还要从右边找到k-index个数
        else if(index > k)//从左边找前K个数
            index = selectK(arr,low,index-1,k);
    }
    return index;
}