P1923 【深基9.例4】求第 k 小的数

// Problem: P1923 【深基9.例4】求第 k 小的数
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P1923
// Memory Limit: 250 MB
// Time Limit: 1000 ms
// User: Pannnn
 
#include <bits/stdc++.h>
 
using namespace std;
 
int getK(vector<int> &info, int l, int r, int k) {
    if (l >= r) {
        return info[r];
    }
    int i = l - 1, j = r + 1;
    int x = info[(l + r) >> 1];
    while (i < j) {
        while (info[++i] < x);
        while (info[--j] > x);
        if (i < j) {
            swap(info[i], info[j]);
        }
    }
    
    if (k <= j) {
        return getK(info, l, j, k);
    } else {
        return getK(info, j + 1, r, k);
    }
}
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    
    int n, k;
    cin >> n >> k;
    
    vector<int> info(n);
    for (int i = 0; i < n; ++i) {
        cin >> info[i];
    }
    cout << getK(info, 0, n - 1, k) << endl;
    return 0;
}