二分查找及简单应用

二分搜索

二分查找也叫二分法或折半查找，每次查找将待分区间划分成两部分并只取一部分。

代码模板

整数二分

1. 可以查找相同数的最大下标
2. 可以查找相同数的最小下标
3. 查找一个数的的位置，以下两种方式都可以

l取mid，r取mid减1，mid需要多加1
r取mid，l取mid加1，mid不变

while(l < r){
	mid = (l + r + 1) >> 1;
	if(check(x)) l = mid;//检查x是否满足所要条件，
	else r = mid - 1;
}
while(l < r){
	mid = (l + r) >> 1;
	if(check(x)) r = mid;//检查x是否满足条件
	else l = mid + 1;
}

浮点数二分

while(l < r){
	mid = (l + r) / 1;
	if(check(x)) r = mid;
	l = mid;
}

例题

AcWing 789 数的范围

给定一个按照升序排列的长度为 n 的整数数组，以及 q 个查询。

对于每个查询，返回一个元素 k 的起始位置和终止位置（位置从 0 开始计数）。

如果数组中不存在该元素，则返回 -1 -1。

输入格式

第一行包含整数 n 和 q，表示数组长度和询问个数。

第二行包含 n 个整数（均在 1∼100001∼10000 范围内），表示完整数组。

接下来 q 行，每行包含一个整数 kk，表示一个询问元素。

输出格式

共 q 行，每行包含两个整数，表示所求元素的起始位置和终止位置。

如果数组中不存在该元素，则返回 -1 -1。

数据范围

1≤n≤100000
1≤q≤10000
1≤k≤10000

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

先构思一下思路，首先找到相同区间的索引最小值，再找相同区间的索引最大值。

1. (索引最小)如果mid >= x，更新右下标，缩小区间为[l,mid],若取到等于，将一直前移，取到所求元素的起始位置；相反mid < x，更新左下标，因为此时没有取到等于，故左区间可以取到(mid + 1)，即缩小区间为[mid+1,r]，一直后移。

2. (索引最大)如果mid<=x，更新左下标，缩小区间为[mid,r]，可以取到索引最大的位置；相反mid > x，更新右下标，同理取 r = mid - 1，一直前移。

   （注意：例子：1，2，2找2 若mid = l + r >> 1的话，第一次mid取到1，在进行第二次查找时，仍然为1，先入了死循环，故将mid改为l + r + 1 >> 1）

要注意mid取法的不同，

代码如下

#include<iostream>

using namespace std;

const int N = 1e6 + 10;
int q[N];

int main()
{	
	int n, m, mid, s;
	cin >> n >> m;
	for(int i = 0; i < n; i++){
		scanf("%d", &q[i]);
	}
	while(m--){
		scanf("%d", &s);
		int l = 0, r = n - 1;
		while(l < r){
			mid = (l + r) >> 1;			
			if(q[mid] >= s) r = mid;
			else l = mid + 1;
		}
		if(q[l] != s) cout << -1 << ' ' << -1 << endl;
		else{
			cout << l << ' ';
			l = 0, r = n - 1;
			while(l < r){
				mid = (l + r + 1) >> 1;
				if(q[mid] <= s) l = mid;
				else r = mid - 1;
			}
			cout << l << endl;
		}
	}
	return 0;
}

浮点数

二分法求开方

求一个数的开方近似数

#include<iostream>

using namespace std;

int main(){
	int n;
	cin >> n;
	double l = 0, r = n, mid;
	while(r - l >= 1e-6){//精度越小越精确
		mid = (l + r) / 2;
		if(n / mid >= mid) l = mid;//没有用mid*mid，避免了数据溢出
		else r = mid;
	}
	cout << l;
	return 0;
}

少用乘法，避免数据溢出