【ybt金牌导航6-1-1】第 k 小数

第 k 小数

题目链接：ybt金牌导航6-1-1

题目大意

求序列中第 k 小的数。
不能 nlogn 做。

思路1（分治）

这道题可以用分治来做，分治有点二分的味道。

它每次选一个数列中的数，然后把比它大的放在前面，比它小的放在后面，然后它放在中间。（这个你可以搞两个指针来搞，每次找到两边然后 swap）
那它的位置就是它的排名（你也只确定了它的，但是够了），然后如果排名就是你要的，就直接记录，退出。
如果排名小了，说明数应该大点，那就处理右边的区间。
如果排名大了，说明数应该小点，那就处理左边的区间。

然后就可以了。

代码1（分治）

#include<cstdio>
#include<algorithm>
#define ll long long

using namespace std;

ll n, a[5000001], k, ans;

ll read() {
	ll re = 0ll;
	char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9') {
		re = re * 10 + c - '0';
		c = getchar();
	}
	return re;
}

void work(ll l, ll r, ll k) {//分治
	if (l >= r) {//确定到点了
		ans = a[l];
		return ;
	}
	
	ll L = l - 1, R = r + 1;
	ll val = a[(l + r) >> 1];
	while (L < R) {//把比它大的放在右边，比它小的放在左边
		while (a[++L] < val);
		while (a[--R] > val);
		
		if (L < R) swap(a[L], a[R]);
	}
	
	if (R - l + 1 >= k) work(l, R, k);//排名大了，要数小点
		else work(R + 1, r, k - (R - l + 1));//排名小了，要数大点
}

int main() {
	n = read();
	k = read();
	for (ll i = 1; i <= n; i++) a[i] = read();
	
	work(1, n, k);
	
	printf("%lld", ans);
	
	return 0;
}

思路2（STL）

STL 自带函数。

STL！！！
永远的神！！！

代码2（STL）

#include<cstdio>
#include<algorithm>
#define ll long long

using namespace std;

ll n, a[5000001], k;

ll read() {
	ll re = 0ll;
	char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9') {
		re = re * 10 + c - '0';
		c = getchar();
	}
	return re;
}

int main() {
	n = read();
	k = read();
	for (ll i = 1; i <= n; i++) a[i] = read();
	
	nth_element(a + 1, a + k, a + n + 1);
	
	printf("%lld", a[k]);
	
	return 0;
}