蓝桥杯备战（AcWing算法基础课）-数的范围（二分）

目录

前言

1 题目描述

2 分析

2.1 第一步

2.2 第二步

3 代码

---

前言

详细的代码里面有自己的理解注释

1 题目描述

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

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

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

输入格式

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

第二行包含 n 个整数（均在 1∼100001∼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

2 分析

按照题目意思可以知道，我们要从输入的升序序列里面找到需要查询的数k的范围，由输入和输出可知，我们需要从左往右看找到第一个>=k，然后从右往左找到<=k第一个的数，即是左右边界，并且由于数组范围和待查询的数据都比较大，暴力肯定会超时，我们知道升序需要有单调性，可以使用二分查找，下面对在1 2 2 3 3 4查找2进行详细分析

2.1 第一步

找左边界，由前面的分析，数组a为1 2 2 3 3 4时，从左往右找第一个>=2的数是

（1）l=0，r=5，mid=2，a[mid]=2，此时因为a是升序的，所以mid=2到r=5的数必定全部>=2，此时mid的位置不一定是第一个>=2的数，那么结果也就是从左往右第一个>=2的数肯定在l到mid之间，所以需要让r=mid

2.2 第二步

找右边界，由前面的分析，数组a为1 2 2 3 3 4时，从右往左找第一个<=2的数是

（1）l=0，r=5，mid=2，a[mid]=2，此时因为a是升序的，所以l=0到mid=2的数必定全部<=2，此时mid的位置不一定是第一个<=2的数（虽然在这里是的，但是如果右边还有一个2就不行），那么结果也就是从右往左第一个<=2的数肯定在mid到r之间，所以需要让l=mid。在这里需要注意让mid=(l+r+1)/2，为什么要+1，是因为最后如果只有两个元素时，l=mid<r，会发生死循环

3 代码

#include<iostream>
#include<algorithm>
#include<vector>
#include<set>
#include<sstream>
#include<cstdio>
#include<string>
#include<cmath>
#include<ctime>

using namespace std;
typedef long long ll;

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

//左边界是从左往右查找到 >= x 的第一个数, 如果没找到返回 > x 的第一个数 
int bsearch_left(int* arr,int l,int r,int num) {
	int mid;
	while(l<r) {
		mid = ( l + r ) >> 1;
		if(arr[mid]>=num) {
			r=mid;
		} else {
			l=mid+1;
		}
	}
	//最后 l == r
	return l;
}
//右边界是从右往左查找到 <= x 的第一个数, 如果没找到返回 < x 的第一个数 
int bsearch_right(int* arr,int l,int r,int num) {
	int mid;
	while(l<r) {
		//凡是涉及左指针移动为 l = mid 的，需要 +1 向上取整
		//因为最后可能出现 [l=mid,r]的区间会进入死循环
		mid = ( l + r + 1 ) >> 1;
		if(arr[mid]<=num) {
			l=mid;
		} else {
			r=mid-1;
		}
	}
	//最后 l == r
	return l;
}
/*
6 3
1 2 2 3 3 4
3
4
5
*/
int main() {
	int n,m;
	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 = bsearch_left(q, 0,n-1,k);
		int r = bsearch_right(q, 0,n-1,k);
		cout<<l<<" "<<r<<"\n";
//		if(q[l]!=k){
//			cout<<"-1 -1\n";
//		}else{
//			cout<<l<<" "<<bsearch_right(q,0,n-1,k)<<"\n";
//		}
	}
	return 0;
}