[AcWing 786] 第k个数

---

点击查看代码

#include<iostream>

using namespace std;
const int N = 1e5 + 10;
int a[N];
int quick_select(int l, int r, int k)
{
    if (l >= r) return a[l];
    int i = l - 1, j = r + 1, x = a[l + r >> 1];
    while (i < j) {
        do i ++;    while (a[i] < x);
        do j --;    while (a[j] > x);
        if (i < j)  swap(a[i], a[j]);
    }
    if (k <= j - l + 1) return quick_select(l, j, k);
    else    return quick_select(j + 1, r, k - (j - l + 1));
}
int main()
{
    int n, k;
    cin >> n >> k;
    for (int i = 0; i < n; i++) cin >> a[i];
    cout << quick_select(0, n - 1, k);
    return 0;
}

---

1. j 指针循环结束后，处在当前数组的第 j - l + 1 位；
2. k > j - l + 1 时，在 [ j + 1, r ] 搜索，原数组第 k 个数是 新一段数组的第 k - ( j - l + 1 ) 个数