从n个数中选出第k大的数——简单随机选择

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <queue>
#include <stack>
#include <set>
#include <iomanip>
using namespace std;
int a[1000005];
int t=0;
long long int quicksort(long long int left,long long int right,int a[])
{
    int x=a[left];
    long long int l=left,r=right;
    while(l<r)
    {
        while(l<r&&x<a[r])r--;
        a[l]=a[r];
        while(l<r&&x>a[l])l++;
        a[r]=a[l];
    }
    a[l]=x;
    return l;
}
long long int solve(long long int l,long long int r,int a[],long long int k)
{
    if(l==r)
        return l;
    long long int pos=quicksort(l,r,a);
    long long int m=pos-l+1;
    if(m==k)
    {
        return pos;
    }
    if(m<k)
        return solve(pos+1,r,a,k-m);
    else
        return solve(l,pos-1,a,k);
}
int main()
{
    long long int n;
    long long int k;
    cin>>n>>k;
    for(long long int i=0; i<n; i++)
    {
        cin>>a[i];
    }
    long long int ans=solve(0,n-1,a,k);
    cout<<a[ans]<<endl;
    return 0;
}