【AcWing】AcWing 786. 第k个数 --C++

本题属于快速选择算法的补充

#include<iostream>
#include<cstring>
using namespace std;
const int D=1e5+10;
int a[D]={0};
int n,k;
void abc(int a[],int t,int m){
    if(t>=m)return;
    int i=t,j=m,tem=a[(t+m)/2];
    do{
        while(a[i]<tem)i++;
        while(a[j]>tem)j--;
        if(i<=j){
            swap(a[i],a[j]);
            i++,j--;
        }
    }while(i<=j);
    abc(a,t,j);
    abc(a,i,m);
}
int main(){
    scanf("%d %d\n",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
        abc(a,1,n);
    printf("%d",a[k]);
}