AcWing 789. 数的范围（C++算法）

AcWing 789. 数的范围

1、题目（来源于AcWing）：
给定一个按照升序排列的长度为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

2、基本思想：

将要二分的区间分为左右两个区域，两个区域满足不同的性质，通过性质不断缩小二分区间，直到找到所找数所在的两个位置。(整数二分要注意处理边界问题）

3、步骤：
①分为两个检查区域
②对于其中一个区域：
mid = (l + r + 1) / 2 (为何+1会在代码后的注意事项中说明）
若满足则缩小区间为 [mid, r] l = mid
若不满足则缩小区间为 [l, mid - 1] r = mid - 1
此时若未找到其中一端则不存在
对于另一个区域：
mid = (l + r + ) / 2
若满足则缩小区间为 [l, mid] r = mid
若不满足则缩小区间为 [mid + 1, r] l = mid + 1

4、C++代码如下（该代码来自AcWing网站）：

#include <iostream>

using namespace std;

const int N = 1e5 + 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--)//用于循环查找m个数
    {
        int x;
        scanf("%d", &x);
        
        int l = 0, r = n-1;
        while(l < r)
        {
            int mid = l + r >> 1;
            if (q[mid] >= x) r = mid;
            else l = mid + 1;
        }//该循环用于查找其中一端
        
        if (q[l] != x) cout << "-1 -1" << endl;//上一个循环结束后l = r,若未找到则不存在
        else
        {
            cout << l << ' ';//先输出上一个循环找出的一端所在位置
            
            int l = 0, r = n-1;
            while(l < r)
            {
                int mid = l + r + 1 >> 1;//注意+1
                if (q[mid] <= x) l = mid;
                else r = mid - 1;
            }//该循环用于查找另一端
            
            cout << l << endl;//输出另一端所在位置
        }
        
    }
    return 0;
}//该代码来自AcWing网站

注意事项：
①第二个 mid = l + r + 1 >> 1 中 mid 要 +1 是因为C++中是向下取整,通过不断缩小范围当 l 与 r 只相差 1 时（l < r）mid=l，则在语句 if (q[mid] <= x) l = mid; 当中 l = mid = l，会导致程序陷入死循环。
②第二次查找时要记得将 l 和 r 的值还原 int l = 0, r = n-1;