二分查找算法模板
题目链接:789 数的范围
题目描述
给定一个按照升序排列的长度为n的整数数组,以及 q 个查询。
对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。
如果数组中不存在该元素,则返回“-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
题解思路(平欢)
二分查找算法思路
整数二分查找的核心是 边界
1、首先找中间值mid ;
2、判断中间值是否满足某种性质(可以将区间的数一分为二的性质)
假设目标值在当前性质中,为性质的最大边缘点(即右端点),那么当满足性质时,mid的值肯定会在目标值的左边,或者就是目标值,接着,将区间长度缩短一半,则目标值所在区间会更新成[mid, r],那么对应更新操作为l = mid; 反之,目标值不在满足性质的区间内,在性质所划分的另一区间则为[l, mid - 1],对应的更新操作是r = mid - 1;此外,需要注意[l, mid - 1]和[mid, r]会不会出现死循环,如果可能出现死循环,那么需要在计算mid时加1;
int bsearch_2(int l, int r)
{
while (l < r)
{
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
假设目标值在当前性质中,为性质的最小边缘点(即左端点),那么当满足性质时,mid的值肯定会在目标值的右边,或者就是目标值,接着,将区间长度缩短一半,则目标值所在区间会更新成[l, mid],那么对应更新操作为r= mid; 反之,目标值不在满足性质的区间内,在性质所划分的另一区间则为[mid + 1, r],对应的更新操作是l = mid + 1;因为[l, mid]和[mid + 1, r]不会出现死循环,那么不需要在计算mid时加1;
int bsearch_1(int l, int r)
{
while (l < r)
{
int mid = l + r >> 1;
if (check(mid)) r = mid;
else l = mid + 1;
}
return l;
}
细节部分
有单调性的一定可以二分,二分可以不需要数据单调性
C++ 代码
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int q[N];
int main()
{
cin >> n >> m;
for (int i = 0; i < n; i ++ ) cin >> q[i];
while (m -- )
{
int x;
cin >> x;
int l = 0, r = n - 1;
while (l < r)
{
int mid = l + r >> 1;
if (q[mid] >= x) r = mid;
else l = mid + 1;
}
if (q[l] != x) cout << "-1 -1" << endl;
else
{
cout << l << ' ';
int l = 0, r = n - 1;
while (l < r)
{
int mid = l + r + 1 >> 1;
if (q[mid] <= x) l = mid;
else r = mid - 1;
}
cout << l << endl;
}
}
return 0;
}
C 代码
#include <stdio.h>
int n, m;
int q[100010];
int main()
{
scanf("%d%d", &n, &m);
for (int i = 0; i < n; i ++ ) scanf("%d", &q[i]);
while (m -- )
{
int x;
scanf("%d", &x);
int l = 0, r = n - 1;
while (l < r)
{
int mid = (l + r) / 2;
if (q[mid] >= x) r = mid;
else l = mid + 1;
}
if (q[l] != x) printf("-1 -1\n");
else
{
printf("%d ", l);
int l = 0, r = n - 1;
while (l < r)
{
int mid = (l + r + 1 ) / 2;
if (q[mid] <= x) l = mid;
else r = mid - 1;
}
printf("%d \n", l);
}
}
return 0;
}