Acwing789. 数的范围

理工大猪猪2023-08-06 22:56

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

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

如果数组中不存在该元素，则返回 -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

cpp 复制代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 100005;
int n, q, x, a[maxn];
int main() {
    cin >> n >> q;
    for (int i = 0; i < n; i++)    
        cin >> a[i];
    while (q--){
        cin >> x;
        int l = 0, r = n - 1;
        while (l < r) {//二分查找的过程
            int mid = l + r >> 1;
            if (a[mid] < x)  l = mid + 1;
            else    r = mid;
        }
        if (a[r] != x) {//此处用x或者y都可以，没有符合条件的点输出-1 -1 并且跳出循环
            cout << "-1 -1" << endl;
            continue;
        }
        //cout << l << endl; l最后总会停在每一种数字的首位上
        int l1 = l, r1 = n;
        while (l1 + 1 < r1) {//把第一次查到的x的值定位，看后面的值还有没有
            int mid = l1 + r1 >> 1;
            if (a[mid] == x)  l1 = mid;
            else    r1 = mid;
            //最终格局是l1 r1紧挨着，然后l1是那个x的最后一个值
        }
        cout << l << " " << l1 << endl;
    }
    return 0;
}