AcWing 786 第k个数

题目描述：

给定一个长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列的第k小的数是多少。

输入格式

第一行包含两个整数 n 和 k。

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

输出格式

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

数据范围

1≤n≤1000001≤n≤100000,
1≤k≤n1≤k≤n

输入样例：

5 3
2 4 1 5 3

输出样例：

3

分析：

本题与AcWing 53 最小的k个数有点类似，不同的是这里只要求求第k小的数，而之前那题要求输出最小的k个数，完全可以采用类似的办法，维护一个大根堆，将数组元素依次加入堆中，一旦堆中元素超过k，就删除堆顶元素，最后留下的那个包含k个元素的大根堆 的堆顶元素就是待求的数。

既然刚总结完快速排序，这里就用快速排序剪枝来做。

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100005;
int a[maxn], n,k;
void quicksort(int *a,int l,int r) {
	if (l >= r)		return;
	int x = a[l],nl = l,nr = r;
	while (l < r) {
		while (l < r && a[r] > x)	r--;
		a[l] = a[r];
		while (l < r && a[l] <= x)	l++;
		a[r] = a[l];
	}
	a[l] = x;
	if(l > k)   quicksort(a, nl, l - 1);
	else if(l < k)  quicksort(a, l + 1, nr);
}
int main() {
	scanf("%d%d", &n,&k);
	for (int i = 1; i <= n; i++)    scanf("%d", &a[i]);
	quicksort(a, 1, n);
	printf("%d\n", a[k]);
	return 0;
}

观察上述代码，可以看见只是在快速排序代码的基础上改动一点点。一轮扫描后，某个元素的最终位置被确定为l，若l < k，则只需要排序l的右边元素，若l > k则只需排序l的左边元素。虽然复杂度是一个级别的，但是足以剪枝掉很多趟扫描。为什么不加上l = k时直接返回呢？一方面，如果加上l = k时直接输出a[k]，然后exit(0)，固然可以，但是一旦排序第k个元素时，直接l >= r被返回掉了，程序不会有任何的输出，只有第k个元素恰好被选为枢轴时才有可能输出；另一方面，当l = k时，左右两边的递归都不会执行，程序会直接返回，某种程度上也不需要加上else来特殊处理下l = k的情况。