二分法的应用 HihoCoder - 1128 二分查找

描述

Nettle最近在玩《艦これ》，因此Nettle收集了很多很多的船(这里我们假设Nettle氪了很多金，开了无数个船位)。去除掉重复的船之后，还剩下N(1≤N≤1,000,000)种不同的船。每一艘船有一个稀有值，任意两艘船的稀有值都不相同，稀有值越小的船越稀有，价值也就越高。
Nettle现在通过大建又造出了一艘船，他想知道这艘船是不是重复的。如果是重复的，那么这艘船在Nettle所有的船里面稀有值排多少位。

问题一

Nettle已经先把自己所有船按照稀有值从小到大排列好了(a[1..N])，我们要做的是看看新得到的船(假设稀有值为K)是否在这个序列中，且有对应的a[i]=K时，i为多少？

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

有序数组长度为N，且满足a[1]&lta[2]<…&lta[N]。
我们随机从数组中选取一个数a[i]，对于需要查找的数K：

若K&lta[i]，则K的下标一定不大于等于i(也有可能K不存在于数组中)；若K=a[i]，则i就是我们要找的答案；若K&gta[i]，则K的下标一定不小于等于K。

利用上面这个性质，我们可以得到这样一个算法：
假定K所在的区间为[L,R]，即a[L]≤K≤a[R]。每次我们从[L,R]中选取一个数Mid，一般情况下选择Mid=(L+R)/2。

若K&lta[Mid]，则更新区间为 R=Mid-1若K=a[Mid]，则Mid为所查询值若K&gta[Mid]，则更新区间为 L=Mid+1
在任意时刻出现L>R时判定K不存在于该数组。

这便是最简单，最基本的二分查找。时间复杂度为O(logN)

问题二

因为Nettle的船太多了，他不愿意去给所有船按照稀有值排序，而是直接告诉了我们每一艘船的稀有值。在这种情况下我们该如何解决这个问题呢？

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

数组长度为N，保证没有重复的数。
一个简单有效的方法是对数组进行排序后使用有序数组的二分查找，时间复杂度为O(NlogN)。

观察我们第一个算法，对于选定的Mid。如果数组满足a[L..Mid-1]&lta[Mid]且a[Mid]&lta[Mid+1..R]，即可进行区间的分割，从而使得区间范围减半。
既然如此，那么我们可以通过一次遍历交换将比a[Mid]小的数放到a[Mid]左边，比a[Mid]大的数放到a[Mid]右边。(这里使用了快速排序的思想)
其他部分仍然同有序数组的二分查找相同，但由于每一次都遍历了整个数组，所以时间复杂度变为：O(N/2^0+N/2^1+N/2^2+1)=O(2N)

由于第一个问题很简单，所以我们就拿第二个问题练练手吧。

输入

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

输出

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

Sample Input

10 5180
2970 663 5480 4192 4949 1 1387 4428 5180 2761

Sample Output

9

代码：

#include<iostream>
#include<algorithm>
using namespace std;

#define MAX_N 1000000
long long a[MAX_N];
int main() {
	
	int n;
	long long k;
	cin >> n >> k;
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	sort(a, a + n);
	
	//二分查找
	int low = 0;
	int high = n-1;
	int mid;
	while (high >= low) {
		mid = (low + high) / 2;
		if (k < a[mid]) {
			high = mid - 1;
		}else if(k == a[mid]) {
			cout << mid+1;
			return 0; 
		}
		else
			low = mid + 1;
	}
	cout << -1;

	return 0;
}