二分查找的关键:有序数组
题目1:数的范围
给定一个按照升序排列的长度为n的整数数组,以及q个查询。对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。如果数组中不存在该元素,则返回-1,-1
输入格式
第一行包含整数n和q,表示数组长度和询问个数。
第二行包含n个整数,表示完整数组。
接下来q行,每行包含一个整数k,表示一个询问元素。
输出格式
共q行,每行包含两个整数,表示所求元素的起始位置和终止位置。
如果数组中不存在该元素,则返回 -1 -1。
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
代码
//整数二分
#include<iostream>
#include<vector>
using namespace std;
//查找区间左端点
int func_left(vector<int>& vec, int key)
{
int start = 0;
int end = vec.size() - 1;
while (start < end)
{
int mid = (start + end) / 2;
if (vec[mid]>=key)
{
end = mid;
}
else
{
start = mid + 1;
}
}
if (vec[start] != key)
{
return -1;
}
return start;
}
//查找区间右端点
int func_right(vector<int>& vec, int key)
{
int start = 0;
int end = vec.size() - 1;
while (start < end)
{
int mid = (start + end + 1) / 2;
if (vec[mid]<=key)
{
start = mid;
}
else
{
end = mid-1;
}
}
if (vec[start] != key)
{
return -1;
}
return end;
}
int main()
{
int n, p;
cin >> n;
cin >> p;
vector<int> vec1;
vector<int> vec2;
for (int i = 0; i < n; i++)
{
int tmp;
cin >> tmp;
vec1.push_back(tmp);
}
for (int i = 0; i < p; i++)
{
int tmp;
cin >> tmp;
vec2.push_back(tmp);
}
for (int i = 0; i < p; i++)
{
cout << func_left(vec1,vec2[i]) << " " << func_right(vec1, vec2[i]) << endl;
}
return 0;
}
浙公网安备 33010602011771号