学习目标:
No.2快速选择算法
学习内容:
快速选择算法的前两步和快速排序一样
走完前两步之后,统计一下x左边的数的个数和x右边数的个数。
如果k比左边数的个数小,那么k一定在x 的左边,所以我们只需要递归排序左边的数即可。
如果k比左边的数大,那么k一定在x的右边,所以我们只需要递归排序右边的数即可。
快排模板:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 100010;
int n, k;
int q[N];
int quick_sort(int l, int r, int 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]);
}
int sl = j - l + 1;
if(k <= sl) return quick_sort(l, j, k);
else return quick_sort(j + 1, r, k - sl);
}
int main()
{
scanf("%d%d", &n, &k);
for(int i = 0;i < n; i ++ ) scanf("%d", &q[i]);
printf("%d", quick_sort(0, n - 1, k));
}