[AcWing 786]第k个数

#include<iostream>
using namespace std;

const int N = 100010;
int n, k;
int q[N];

int quick_sort(int l, int r, int k){
    if(l == r) return q[l];
    
    int x = q[l];
    int i = l-1;
    int j = r+1;
    while(i < j){
        while(q[++i] < x);
        while(q[--j] > x);
        if(i < j) swap(q[i], q[j]);
    }//比x小的数放在左边，比x大的数放在右边
    
    int sl = j-l+1; //左边的数的个数
    if(k <= sl) return quick_sort(l, j, k);//总区间上的第k个数在左边的第k个位置上
    
    return quick_sort(j+1, r, k-sl);//总区间上的第k个数在右边的第k-sl个位置上

}

int main(){
    cin >> n >> k;
    for(int i = 0; i < n; i++)   cin >> q[i];
    
    cout << quick_sort(0, n-1, k);
    
    return 0;
}