acwing 789. 数的范围

2022-01-03 04:39:22

789. 数的范围

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

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

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

输入格式

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

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

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

输出格式

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

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

数据范围

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

 1 #include <iostream>
 2 using namespace std;
 3 const int N = 1e6 + 10;
 4 int q[N], n, m, k;
 5 
 6 int main()
 7 {
 8     scanf("%d %d", &n, &m);
 9     for(int i = 0; i < n; i ++) scanf("%d", &q[i]);
10     while(m --)
11     {
12         scanf("%d", &k);
13         int l = 0, r = n - 1;
14         while(l < r)
15         {
16             int mid = (l + r) >> 1;
17             q[mid] >= k ? r = mid :l = mid + 1;
18         }
19         if (q[l] != k) printf("-1 -1\n") ;
20         else
21         {
22             printf("%d ", l);
23             int l = 0, r = n - 1;
24             while (l < r)
25             {
26                 int mid = (l + r + 1) >> 1; 
27                 q[mid] <= k ? l = mid : r = mid - 1;
28             }
29             printf("%d\n", l);
30         }
31     }
32     return 0;
33 }