[AcWing] 数的范围

数的范围

题目

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

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

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

输入格式

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

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

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

输出格式

共q行，每行包含两个整数，表示所求元素的起始位置和终止位置。

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

数据范围

$$\begin{gathered} 1\le n\le 100000 \\ 1\le q\le 10000 \\ 1\le k\le 10000 \end{gathered}$$

输入样例

6 3
1 2 2 3 3 4
3
4
5

输出样例

3 4
5 5
-1 -1

题解

思路

• 二分法确定左右边界

#include <iostream>

using namespace std;

const int N = 1e5 + 20;
int n, m, x, a[N];

int main () {
    cin >> n >> m;
    for (int i = 0; i < n; i ++) cin >> a[i];
    
    for (int i = 0; i < m; i ++) {
        cin >> x;
        
        int l = 0, r = n - 1;
        while (l < r) {
            int mid = l + r >> 1;
            if (a[mid] >= x) r = mid;
            else l = mid + 1;
        }
        
        if (a[r] != x) {
            cout << "-1 -1" << endl;
            continue;
        }
        
        cout << r << " ";
        
        l = 0, r = n - 1;
        while (l < r) {
            int mid = l + r + 1>> 1;
            if (a[mid] <= x) l = mid;
            else r = mid - 1;
        }
        cout << r << endl;
    }
    
    return 0;
}