[AcWing]786. 第k个数（C++实现）

1. 题目

2. 读题（需要重点注意的东西）

思路：使用快排+剪枝的策略，将时间复杂度简化到O(n)
剪枝的思想：要求第k小的数。在第一次快排后，分界点左边的数都比分界点右边的数小，左边有j - l + 1个数，如果k <= j - l + 1，则k一定在左边，因此只需要对左边再进行快排即可，即quick_sort(q, l, j, k)；如果k > j - l + 1，则k在右半部分，且k为右半部分第k - (j - l + 1)个数，即quick_sort(q, j+1, r, k - (j - l + 1))。

3. 解法

---------------------------------------------------解法---------------------------------------------------：

#include <iostream>

using namespace std;

const int N = 1000010;

int q[N];

int quick_sort(int q[], int l, int r, int k)
{
	// 区间只有一个元素，且k在区间中，则k即为该元素，直接返回即可
    if (l == r) return q[l];

    int i = l - 1, j = r + 1, x = q[l + 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]);
    }

    if (j - l + 1 >= k) return quick_sort(q, l, j, k);
    // // 除去了j-l+1个比k小的数，那么k是剩下数中的第 k - (j-l+1)大的数
    else return quick_sort(q, j + 1, r, k - (j - l + 1));
}

int main()
{
    int n, k;
    cin >> n >> k;

    for (int i = 0; i < n; i ++ ) scanf("%d", &q[i]);

    cout << quick_sort(q, 0, n - 1, k) << endl;

    return 0;
}

背模板的时候记住判断条件：if (j - l + 1 >= k)

4. 可能有帮助的前置习题

• [AcWing]785. 快速排序（C++实现）

5. 所用到的数据结构与算法思想

6. 总结

快速选择算法，快排模板的应用

这里求的是第k小的数
如果要求第k大的数，只需要将k = n - k + 1即可。