AcWing 786. 第k个数


#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int q[N];
void quick_sort(int q[],int l,int r){
    if(l>=r) return;
    int x=q[l+r+1 >>1];
    int i=l-1,j=r+1;
    while(i<j){
        do i++;while(q[i]<x);
        do j--;while(q[j]>x);
        if(i<j) swap(q[i],q[j]);
    }
    quick_sort(q,l,i-1);
    quick_sort(q,j+1,r);
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
        scanf("%d",&q[i]);
    quick_sort(q,0,n-1);
    printf("%d",q[m-1]);
    return 0;
}