快速选择算法-基于快排

快速选择算法

快速选择，就是在一个无需的数组中查找第k个小的数。我们可以很快的想到可以先将数组排序在进行选择，但时间复杂度为$O(n\log n)$。这时我们就可以适用快速选择算法优化，时间复杂读为$O(n)$。

例题

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。

输入格式
第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整数数列。

输出格式
输出一个整数，表示数列的第 k 小数。

数据范围
1≤n≤100000,
1≤k≤n

题目链接

算法思路

首先改算法的思想是基于 快排 分治的思想，在我们进行快排操作时我们只需要对我们需要的部分进行递归计算即可。大概思路如下：
1.出口条件将原来的 l>=r ，该为 l==r 并进值返回。
2.对 l 到 r 区间进行快排操作，注意分界条件，与快排相同。
3.在递归调用时 我们不需要对左右两边都进行调用，只需要判断我们选择的数k在左边还是在右边，然后只调用k所在的区间即可。

时间复杂度：$O(n)$

代码实现

#include <bits/stdc++.h>
using namespace std;

int q[100010];

int quick_select(int l, int r, int k)
{
    if (l == r) return q[l+r>>1];

    int i = l - 1, j = r + 1;
    int 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 (k <= j) return quick_select(l, j, k);
    return quick_select(j + 1, r, k);
}

int main()
{
	int n, k;
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i ++ ) scanf("%d", &q[i]);
	
	printf("%d",quick_select(0, n-1, k-1));
    return 0;
}