求第k小的数

洛谷 P1923 求第k小的数（普及/提高-）（k从0开始取）

看到这题第一反应，直接 sort() 再输出 a[k-1] 就完事了，觉得太偷懒就没这么做（真的这么做也会超时，因为只需要求第 k 小的数，sort() 则给每一个数都排了序）

虽然已知这个题是分治，但是怎么分治？

如果是求最小的数，那么完全可以二分，将左边的最小值与右边的最小值进行比较，得到最小值。

因为本题是求第k小的数，觉得可以二分后分别保留左、右的 k 个最小的数，再比较以合并结果，但是很难实现，因为太复杂了。

于是写了第一版朴实的代码，只维护 k 个最小的数，复杂度为O(k*n)，数据量较小时没问题：

int res[k + 1];
	fill(res, res + k + 1, INT_MAX);
	int temp;
	for (int i = 0; i < n; i++) {
		cin >> temp;
		if (temp >= res[k])continue;
		//若当前数需要插入数组
		for (int j = 0; j <= k; j++) {
			if (temp < res[j]) {
				for (int m = k - 1; m >= j; m--) {
					res[m + 1] = res[m];
				}
				res[j] = temp;
				break;//只需一次调整即可，防止覆盖后续的数
			}
		}
	}
	cout << res[k];

但是 k 一旦变大，时间复杂度立刻就上来了。这时才知道，这里的分治法核心是快排+剪枝：每次确定一个数的位置（它左边的数均不大于它，右边的数均不小于它），然后将这个位置与 k 进行比较，若正好是 k ，那么该数就是答案；若小于 k ，说明应去右侧找 k（递归）；大于 k 时去左侧。

由于平均情况下，每次可以有一半的数据不需处理（只需去一侧找 k ），因此降低了时间复杂度。

这样就有了正确的代码：

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

int x[5000000];
int k;

int quicksort(int l, int r) {
	//快速排序
	int mid = x[l];
	//取出当前区间最左端的值为中值(标的值)，这样还可以使得x[l]可以被覆盖，统一了后续操作
	while (l < r) {
		while (l < r && x[r] >= mid)r--;
		x[l] = x[r];
		while (l < r && x[l] <= mid)l++;
		x[r] = x[l];//若有l赋给r的动作，则和上述r赋给l的动作一定是成对出现的，保证了正确性
	}
	//调整结束，此时l==r且左侧均<=mid，右侧均>=mid
	x[l] = mid;
	return l;
}

int search(int l, int r) {
	//分治法的核心部分
	//由于每次只需要处理接近一半的数据，因此时间复杂度低
	int m = quicksort(l, r);
	if (m == k)return x[m];
	if (m < k)return search(m + 1, r);
	if (m > k)return search(l, m - 1);
}

int main() {
	//求n个数中第k小的数，k从0开始
	//n为奇数
	int n; cin >> n >> k;
	for (int i = 0; i < n; i++)scanf("%d", &x[i]);
	cout << search(0, n - 1);
}