基础算法(3) —— 整数二分模板

整数二分

用二分的思想可以找出这个分界点。
假设我们想二分出来红色（不满足条件的范围）这边的分界点：

1. 找一个中间值mid ：l+r >> 1
2. 写一个函数check()判断时mid时在满足条件（绿色）范围内，还是在不满足条件（红色）范围内。
3. （注意：这里为什么是 l+r+1 ）
   For example：
   如果 l = r - 1 (即左右边界只相差1) ，假设为l+r >> 1 (向下取整)假设if(check(mid)) 判断为true （此时mid = l + r >> 1 == l）则执行l=mid; l=mid=l;陷入死循环，左右边界一直为[l, r] 没有变
   如果是l+r+1>>1则不会发生这个问题（详情自己参考上面推导一遍即可）。
   

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

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

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

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

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

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

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

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

数据范围
$1\le n\le 100000$
$1\le q\le 10000$
$1\le k\le 10000$
输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

代码：

#include<iostream>
#include<cstdio>
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)

const int N = 100005;

int n, m;
int q[N];
using namespace std;

/———————————————————————————————— 整数二分模板 ————————————————————————————————\
bool check(int x) {/* ... */ } // 检查x是否满足某种性质

// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用：
int bsearch_1(int l, int r)
{
	while (l < r)
	{
		int mid = l + r >> 1;
		if (check(mid)) r = mid;    // check()判断mid是否满足性质
		else l = mid + 1;
	}
	return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用：
int bsearch_2(int l, int r)
{
	while (l < r)
	{
		int mid = l + r + 1 >> 1;
		if (check(mid)) l = mid;
		else r = mid - 1;
	}
	return l;
}
\———————————————————————————————— 整数二分模板 ————————————————————————————————/
int main()
{
	IOS;
	cin >> n >> m;
	for (int i = 0; i < n; ++i)
		cin >> q[i];
	while (m--)
	{
		int x;//分界点
		cin >> x;
		int l = 0, r = n - 1;//设定边界
		while (l < r)
		{
			int mid = l + r >> 1;
			if (q[mid] >= x)//说明mid一定在右半边，则边界一定在左边
				r = mid;//区间从[l, r]更新为[l, mid]
			else
				l = mid + 1;//因为是整数二分，所以不能mid重合，则一定是mid + 1，将区间更新为[mid + 1, r]

			//当序列中不存在x的话，二分出来的结果是：最前面那个≥x的数（序列从左往右看第一个满足≥x的数）。
			if (q[l] != x)
				printf("-1 -1\n");
			else
			{
				printf("%d ", l);//这里输出l和r是一样的，因为二分结束之后 l==r。

				int l = 0, r = n - 1;
				while (l < r)
				{
					int mid = l + r + 1 >> 1;
					if (q[mid] <= x)//说明mid一定在左半边，则边界一定在右边或者mid就是边界
						l = mid;//区间从[l, r]更新为[mid, r]
					else
						r = mid - 1;//将区间更新为[l, mid - 1]
				}
				printf("%d\n", l);
			}
		}
	}
	return 0;
}