二分算法模版

今天，我详细的学习了二分算法，并整理了算法的模版，先以一个例题，来引入二分算法模版。

数的范围问题

题目：

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

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

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

输入格式

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

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

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

输出格式

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

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

数据范围

1≤n≤1000001≤100000
1≤q≤100001≤10000
1≤k≤100001≤10000

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

题目分析：

这个问题我们应该怎样分析呢？

首先，我们看到了这个数组是一个单调的，也就是说，是一个拍好顺序的数组，那么此时我第一个想到的就是用二分办法来解决这个问题。我们可以先用二分，先找到一个边界，再使用二分找到另外一个边界，这样就可以找到目标数的范围了。话不多说，开始撸代码~。

#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int n, m;
int q[N];
int main(){
	scanf("%d%d", &n,&m);
	for(int i = 0; i < n; i ++)scanf("%d",&q[i]);
	while(m --){
		int k;
		scanf("%d",&k);
		int l = 0, r = n - 1;
		while(l < r){
			//在这里我们先寻找一个左边界 
			//这里要不要  +1 后面再分析 
			int mid = l + r >> 1;
			if(q[mid] >= k) r = mid;//因为左边界的右侧都是大于等于 k 的 
			else l = mid + 1; 
		}
		if(q[l] != k) cout << "-1 -1" << endl;//没有结果 
		else{
			cout << l <<' ';
			int l = 0, r = n - 1;
			while(l < r){
				int mid = l + r + 1 >> 1;
				if(q[mid] <= k) l = mid;//因为有边界的左侧都是小于等于 k 的 
				else r = mid - 1;
			}
			cout<< l << endl;
		}
	}
	return 0;
}

当我把这个算法写出来之后，大家可能会发现一个问题。

第一个二分中mid 的计算方法是这样的

int mid = l + r >> 1;

第二个二分中的mid的计算方法确是这样的

int mid = l + r + 1 >> 1;

这到底是为什么呢？

其实，这是一个二分中一个非常细节的问题，如果不注意的话，很容易进入死循环。我先把二分的代码放到下面，我们再来解释这个问题。

//区间[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;
        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;
}

我们可以假设

r = l + 1 mid = l 

那么 如果check = true 也就是说，l  =  l 这样的话，就会进入死循环