蓝桥杯算法笔记＜1＞二分

二分算法的例题

---

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

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

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

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

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

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

输出格式
共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

---

二分是一个基础算法，但是如果掌握不好，很容易写出各种各样的死循环
这道题是二分算法的一个模板题
我的代码如下：

#include<cstdio>
#include<iostream>
using namespace std;
const int N = 100010;

int q[N];

int main()
{
	
	int n, m;
	cin >> n >> m;
	for (int i = 0; i < n; i++)
	{
		scanf("%d", &q[i]);
	}

	int input;
	for (int i = 0; i < m; i++)
	{
		cin >> input;

		/*开始对左端点进行二分*/
		int l = 0, r = n - 1;
		int mid;
		while (l < r)
		{
			mid = (l + r) / 2;
			
			if (q[mid] < input)
				l = mid+1;
			else if (q[mid] >= input)
				r = mid;
		}

		if (input == q[l])
			cout << l << " ";
		else
			cout << "-1 ";

		/*开始对右端点进行二分*/
		l = 0, r = n - 1;
		while (l < r)
		{
			mid = (l + r + 1) / 2;
			
			if (q[mid] <= input)
				l = mid;
			else if (q[mid] > input)
				r = mid-1;
		}
		if (input == q[r])
			cout << r << endl;
		else
			cout << "-1" << endl;


	}


	return 0;

}

这道题体现了二分算法中两个需要特别注意的地方

1.进入while循环时，mid赋值为 (l+r)/2 还是 (l+r+1)/2 （即选取二分mid时，到底时向上取整还是向下取整的问题）

2.由于二分的区间是闭区间，所以在划分区间的时候可以划分为[l,mid][mid+1,r]或者[l,mid-1] [mid,r]

---

1.设想数组中只有1和2 此时l=0，r=1，若使用向上取整，则mid=1，若使用向下取整，mid=0
如果希望通过二分获取元素1的下标，就必须要使mid=0，这样才能让r赋值为mid，从而避免死循环

同理如果希望通过二分获取元素2的下标，就必须要mid=1

总结下来，如果希望得到的下标是左区间的右端点，就需要使用mid=(l+r+1)向上取整
如果希望得到的是右区间的左端点，就需要使用mid=(l+r)/2进行向下取整

---

2.如果希望获取的是左区间的右端点，就需要在划分区间时划分为[l,mid] 和[mid+1,r]
如果希望获取的是右区间的左端点，就需要划分为[l,mid-1]和[mid,r]