[排序] 第k个数(模板+快速选择算法)

0. 前言

快速选择算法的思想还是很厉害的。是快速排序的应用，一般将一次快排称为做了一次 partition 操作。c++ stl 也将该函数纳入其中，感兴趣可取查看相关 API。

1. 快速选择算法

786. 第k个数

本质： 快排应用

时间复杂度：$O(n)$
空间复杂度：$O(logn)$，基于递归，用到栈空间

思路：

• 一次快排，数轴左边都 <=x，数轴右边都 >= x
• 左边的元素个数为 s = j - l + 1
• 如果 k <= s ，说明第 k 个数还在左半区间，则下次就递归左半区间
• 否则，递归右半区间的时候，寻找右半区间的第 k-s 个数即可

注意： 这里是 s = j - l + 1，而不是 s = i - l，因为 while 循环结束，i,j 可能相等，可能穿越。区间就被分成 [l,j] 为左半边区间，[i,r] 为右半边区间。其中 [l, j] 是小于等于基准值的数，右半边是大于等于基准值的数。

模板代码：

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5+5;

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+r>>1], 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]);
    }
    int s = j - l + 1;
    if (s >= k) return quick_sort(l, j, k);
    return quick_sort(j + 1, r, k - s);
}

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