给定一个按照升序排列的长度为 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
上代码:
#include<iostream>
using namespace std;
const int N=100010;
int n,q,k,a[N];
int main()
{
cin>>n>>q;
for(int i=0;i<n;i++)
cin>>a[i];
while(q--)
{
cin>>k;
int l=0,r=n-1;
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=0,r=n-1;
while(l<r)
{
int mid= l+r+1 >> 1;
if(a[mid]<=k) l=mid;
else r=mid-1;
}
cout<<l<<endl;
}
}
return 0;
}
二分法是一个比较绕的的算法,思路很简单但是写起来总是写不对,今天我们通过这一道题来简单了解一下二分法。
情况一:
首先,我们要意识到二分法并不只适用于升序降序,它实质上是一个寻找边界的算法,只不过相较于其他序列,升序降序的序列更容易设置一个判断标准,我们通过这个标准来判断此序列的中点在左右哪一边(左边不符合这个标准,右边符合这个标准),如果在右边,说明mid(中间数)符合这一标准,我们所要寻找的边界可能在mid左边,也可能就是mid,于是我们把查找范围更新l到mid,如果mid在左边,说明这个序列中间的这个数并不符合这一标准,我们所要寻找的边界一定在mid的右边,于是把范围变成从mid+1到r,如此不断更新直到l等于r,此时l与r都指向了我们所要找的边界。
情况二:
还有另一种情况是,如果我们设置的标准左边是满足这个标准的,右边是不满足这个标准的,但是这个标准并不是我们想要的结果,比如说这道题我们要查找数字在数列中的最后一个位置,设置的标准是mid是否小于等于这个数,由于这个数列是从小到大排序的,因此左边的数列一定是符合mid小于等于k的,而我们需要找的并不是这个条件,而是要找大于k的那个临界值,这样的情况又和上面讲的情况有所不同。情况一样,我们首先判断mid是否满足这个条件,如果满足这个条件说明我们要寻找的临界值可能是mid,也有可能在mid右边,于是l=mid,如果mid不满足这个条件,说明我们要找的临界值在mid左边,所以r=mid-1。
情况一中,mid=l+r/2,情况二中,mid=l+r+1/2,这是因为满足标准的情况下,情况一是使右边界变化,情况二是使左边界变化,如果刚开始的l=0,r=1,情况二如果也是mid=l+r/2,刚开始mid=0,满足条件的情况下l=mid=0,没有变化,会一直循环下去,所以我们要在这里加以区分。
还是老样子上图解吧!针对这道题。
此时mid=l+r>>1=2,此时的标准是a[mid]是否大于等于3,很明显不满足标准,于是l=mid+1=3
此时mid=4,a[mid]大于3,于是r=mid=4
此时mid=3,a[mid]等于3,于是r=mid=3,此时l=r退出循环,由于a[l]=3,于是我们输出此时的l,即3在这个数列中第一次出现的位置。
我们再判断3在这个数列中最后一次出现的位置,一夜回到解放前我们再来一遍哈
虽然位置还是那么个位置,但是条件已经不是那个条件了,a[mid]等于3,改变不再是r而是l,l=mid=3
a[mid]等于3,所以还是l变
此时a[mid]>3,于是r=mid-1=4,此时l=r,它们指向的位置就是3最后出现的位置。
我们得到了正确的结果。
注意,当l与r重叠时,l与r指向的一定是边界值,如果l与r指向的数不等于我们所要寻找的数,那一定就意味着数组中没有这个数,我们直接输出(-1,-1)就可以。