【二分查找】数的范围

思路

题目主要的意思在一个升序也就是有序的数组中 找到一段相同数组的起始位置和最终位置 也就是查找两个数的问题

题目给的是升序数列让人很容易想到二分查找来查找两个数

二分查找

我们此时的问题就转化为查找到这段序列的起始位置和终止位置 我们可以用二分查找的方式

左边界

我们可以先找这段序列的左边界

例如下面这段序列

我们要找到箭头指向的这个数的位置

我们要设置两个标记来标记每一次二分的左右边界
第一次二分定义 （l = 0 r = n-1） n为序列的长度
在设置一个 mid 变量来标记序列中间位置

每次二分检查 mid 和 我们要找的这个数的大小关系
直到l >= r此时就找到了左边界

int left_bound(int l, int r) {
    while (l < r) {
        int mid = l + r >> 1;//求中间值
        if (q[mid] >= k) r = mid;//更新有边界
        else l = mid + 1;//更新左边界
    }
    return l;
}

我们在这里要尤其注意边界问题 因为这一段序列中有重复数据

在查找左边界时 如果 q[mid]>=k 条件成立 说明左边界点不在右半部分 此时要更新右边界的值
但是存在这个点就是我们要找的点的情况 此时mid点就不可以跳过 在更新边界点时就要保留mid点 就是 r = mid

但是如果 q[mid]>=k条件不成立 这个时候说明左边节点 在右半部分 此时要更新左半部分
此时这个点不可能是我们要找的点 因为这个点必小于k 所以将l更新为mid+1跳过这个点

右边界

然后来找这段数的右边界

还是这段序列 此时我们要找的是箭头指向的这个数据 如果理解了找左边界的情况 这个情况也类似

还是
我们要找到箭头指向的这个数的位置
我们要设置两个标记来标记每一次二分的左右边界
第一次二分定义 （l = 0 r = n-1） n为序列的长度
在设置一个 mid 变量来标记序列中间位置

每次二分检查 mid 和 我们要找的这个数的大小关系
直到l >= r此时就找到了右边界

int right_bound(int l, int r) {
    while (l < r) {
        int mid = l + r + 1 >> 1;
        if (q[mid] <= k) l = mid;
        else r = mid - 1;
    }
    return l;
}

还是和找左边界一样的考虑方法

在查找右边界时 如果 q[mid] <= k 条件成立 说明右边界点不在左半部分 此时要更新左边界的值 但是存在q[mid] = k 的情况 所以可能会出现mid位置正好是我们要的右边界 所以此时 保留mid位置的值 l = mid

如果q[mid] <= k条件不成立 说明右边界点不在右半部分 此时要更新右边界的值 由于此时不可能存在q[mid] = k 的情况出现 所以mid位置的值不需要保存 r = mid - 1

注意再求右边界的时候为什么 mid = (l + r + 1) / 2
我们考虑这种情况 在求右边界时 右边界恰好等于 l + 1 的情况 此时mid 如果等于(r + l)/2 则带入 r = l +1则有 mid = (l + l + 1) / 2 = l 此时q[mid] <= k 成立 将mid值赋给l
我们发现在一次二分后 我们的区间还是[l,r] 并没有发生改变

但是我们+1后 mid = (l + l + 1 + 1) / 2 = l + 1 此时 mid指向的值恰好为右边界 满足q[mid] <= k 条件 将此时 l = mid = l + 1 = r 区间更新为 [r, r]就没有问题

代码实现

# include<iostream>
using namespace std;

const int N = 100010;
int q[N];
int n, m;

int main()
{
	scanf("%d%d", &n, &m);
	for (int i = 0; i < n; i++)
	{
		scanf("%d", &q[i]);
	}
	while (m--)
	{
		int k = 0;
		scanf("%d", &k);
		int l = 0, r = n - 1;
		while (l < r)
		{
			int mid = l + r >> 1;//求mid
			if (q[mid] >= k)//找左边界
				r = mid;
			else
				l = mid + 1;
		}
		if (q[l] != k)//找不到左边界 说明没有这个数据 返回 -1
			printf("-1 -1\n");
		else
		{
			printf("%d ", l);//输出左边界
			int l = 0, r = n - 1;
			while (l < r)//找右边界
			{
				int mid = l + r + 1 >> 1;
				if (q[mid] <= k)
					l = mid;
				else
					r = mid - 1;
			}
			printf("%d\n", l);//输出右边界
		}
	}

	return 0;
}