一个无序数组中第K大的元素

int partition(int *arr, int n){
    int index = rand()%n;
    swap(arr[index], arr[n-1]);
    int i = 0;
    int j = 0;
    for(;j < n; j++)
    {
        if(arr[j] < arr[n-1]){
            swap(arr[i++], arr[j]);
        }
    }
    swap(arr[i],arr[n-1]);
    return i;
}

int getMaxK(int * arr, int n , int k){
    
    if(k < 0) return -1;
    if(n < k) return -1;
    int mid = partition(arr,n);
    if(mid == n-k) return arr[mid];
    else if(mid < n-k)
        // n-k 大于mid 表示要找的数在mid 的右半部份， 所以从原数组的mid+1 开始找。 右半部份的个数是n-(mid+1) 还是在第k个元素
        return getMaxK(arr+mid+1, n-(mid + 1), k);
    else
        // n-k 小于mid 表示要找的数在mid的 左半部分， 所以从原数组开始找 个数是mid 找 k-(n-mid)
        return getMaxK(arr, mid, k-(n-mid));
}


int main()
{
    int arr[] = {10,1,7,2,5,6,8,3,4,9};
    cout <<getMaxK(arr, 10, 3);
    cout << endl;

}