二分·二分查找之k小数

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

在上一回里我们知道Nettle在玩《艦これ》，Nettle的镇守府有很多船位，但船位再多也是有限的。Nettle通过捞船又出了一艘稀有的船，但是已有的N(1≤N≤1,000,000)个船位都已经有船了。所以Nettle不得不把其中一艘船拆掉来让位给新的船。Nettle思考了很久，决定随机选择一个k，然后拆掉稀有度第k小的船。 已知每一艘船都有自己的稀有度，Nettle现在把所有船的稀有度值告诉你，希望你能帮他找出目标船。

提示：非有序数组的二分查找之二

输入

第1行：2个整数N,k。N表示数组长度，
第2行：N个整数，表示a[1..N]，保证不会出现重复的数，1≤a[i]≤2,000,000,000。

输出

第1行：一个整数t，表示t在数组中是第k小的数，若K不在数组中，输出-1。

样例输入

10 4
1732 4176 2602 6176 1303 6207 3125 1 1011 6600

样例输出

1732

思路：
数组长度为N，保证没有重复的数。查找第k小的数
一个简单有效的方法是对数组进行排序后输出第k位，时间复杂度为O(NlogN)。
这里需要对“非有序数组的二分查找”进行改进。
同样，对于选定的Mid。如果数组满足a[L..Mid-1]<a[Mid]且a[Mid]<a[Mid+1..R]，即可进行区间的分割，从而使得区间范围减半。
每一次我们随机从数组中选取一个数m，将比m小的数从a[1]开始放置，将比m大的数从a[N]开始放置。当其他所有的数都放置完成后。最后剩下的格子放置m，不妨设为a[Mid]， 则在我们进行一次遍历交换以后有如下性质：
若k<Mid，则有第k小的数一定在a[L..Mid-1]之中，令R = Mid - 1
若k=Mid，则m就是我们要找的数
若k>Mid，则第k小的数一定在a[Mid+1..R]之中，令L = Mid + 1。此处需特别注意，在a[Mid+1..R]我们查找的k'和原来的k不相同，现在的k'=k-Mid。
这样重复LogN次之后便可找到目标数。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int num[1000000];
int bSearch(int low, int high, int target)
{
	if (low > high)
		return -1;
	int mid = num[(low + high) / 2];
	int left = low;
	int right = high;
	//这里只能针对非重复数数组
	while (left < right)
	{
		while (left < right && num[left] < mid)//这里语句可以替换为while(num[left] < mid)
			++left;
		while (right > left && num[right] > mid)//这里可以替换为while(num[right]>mid)
			--right;
		swap(num[left], num[right]);
	}
	num[left] = mid;//这句不需要
	if (left == target)
		return mid;
	else if (left < target)
		return bSearch(left + 1, high, target);
	else 
		return bSearch(low, left - 1, target);
}
int main(void)
{
	int n, k;
	scanf("%d%d", &n, &k);
	for (int i= 0; i < n; ++i)
		scanf("%d", &num[i]);
	printf("%d\n", bSearch(0, n-1, k - 1));
	return 0;
}
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int num[1000000];
int bSearch(int low, int high, int target)
{
	if (low > high)
		return -1;
	int mid = num[low];
	int left = low;
	int right = high;
	//这里只能针对重复数数组
	while (left < right)
	{
		while (right > left && num[right] >= mid)
			--right;
		swap(num[right], num[left]);
		while (left < right && num[left] <= mid)
			++left;
		swap(num[right], num[left]);
	}
	if (left == target)
		return mid;
	else if (left < target)
		return bSearch(left + 1, high, target);
	else 
		return bSearch(low, left - 1, target);
}
int main(void)
{
	int n, k;
	scanf("%d%d", &n, &k);
	for (int i= 0; i < n; ++i)
		scanf("%d", &num[i]);
	printf("%d\n", bSearch(0, n-1, k - 1));
	return 0;
}