- aeo_lian 的博客
数的范围(二分查找)
- 2024-10-3 15:55:56 @
数的范围
给定一个长度为 n 的整数数组,以及 q 个查询。 对于每个查询,返回按升序排序后的数组的一个元素 k 的起始位置和终止位置(位置从 1 开始计数)。
如果数组中不存在该元素,则返回 -1 -1
。
输入格式
第一行包含整数 n 和 q,表示数组长度和询问个数。
第二行包含 n 个整数(均在 1 ∼ 10000 范围内),表示完整数组。
接下来 q 行,每行包含一个整数 k ,表示一个询问元素。
输出格式
共 q 行,每行包含两个整数,表示所求元素的起始位置和终止位置。
如果数组中不存在该元素,则返回 -1 -1
。
输入样例
6 3
1 2 2 3 3 4
3
4
5
输出样例
输出数据 1
4 5
6 6
-1 -1
数据范围
1≤n≤100000
1≤q≤10000
1≤k≤10000
模板
#include<bits/stdc++.h>
using namespace std;
int n, k, q, a[100010];
int main()
{
cin >> n >> q;
for(int i = 1; i <= n; i++)
{
cin >> a[i];
}
for(int i = 1; i <= q; i++)
{
cin >> k;
int l = 1, r = n;
while(l < r)
{
int mid = l+r>>1;
if(a[mid] >= k)
{
r = mid;
}
else
{
l = mid + 1;
}
}
if(a[l] != k)
{
cout << "-1 -1" << endl;
}
else
{
cout << l << " ";
int l = 1, r = n;
while(l < r)
{
int mid = l+r+1>>1;
if(a[mid] <= k)
{
l = mid;
}
else
{
r = mid - 1;
}
}
cout << l << endl;
}
}
return 0;
}