今天,咱们来谈谈两种简单的二分搜索
第一个:从小到大排序的int数组a里查找元素p,如果找到,则返回元素下标,如果找不到,则返回-1.
int BinarySearch(int a[],int size,int p)
{
int l=0;
int r=size-1;
while(l<=r)
{
int mid=l+(r-l)/2;
if(p==a[mid])
return mid;
else if(p>a[mid])
l=mid+1;
else
r=mid-1;
}
return -1;
}
第二个:
从小到大排序的int数组a里查找比给定整数p小的,下标最大的元素。若找到则返回起下标,找不到则返回-1
int LowerBound(int a[],int size,int p)
{
int l=0;
int r=size-1;
int ans=-1;
while(l<=r)
{
int mid=l+(r-l)/2;
if(a[mid]>=p)
r=mid-1;
else
{
ans=mid;
l=mid+1;
}
}
return ans;
}
来个具体的例子瞅一下吧!!!
**01:查找最接近的元素 **
描述
在一个非降序列中,查找与给定值最接近的元素。
输入
第一行包含一个整数n,为非降序列长度。1 <= n <= 100000。
第二行包含n个整数,为非降序列各元素。所有元素的大小均在0-1,000,000,000之间。
第三行包含一个整数m,为要询问的给定值个数。1 <= m <= 10000。
接下来m行,每行一个整数,为要询问最接近元素的给定值。所有给定值的大小均在0-1,000,000,000之间。
输出
m行,每行一个整数,为最接近相应给定值的元素值,保持输入顺序。若有多个值满足条件,输出最小的一个。
样例输入
3
2 5 8
2
10
5
样例输出
8
5
#include<iostream>
#include<cmath>
using namespace std;
int f(int a[],int size,int p)
{
int l=0;
int r=size-1;
int ans=-1;
while(l<=r)
{
int mid=l+(r-l)/2;
if(a[mid]>p)
r=mid-1;
else if(a[mid]==p)
return mid;
else
{
ans=mid;
l=mid+1;
}
}
return ans;
}
int main()
{
int k,n,m,a[100001],b[10001],i,j;
cin>>n;
for(i=0;i<n;i++)
cin>>a[i];
cin>>m;
for(i=0;i<m;i++)
cin>>b[i];
for(i=0;i<m;i++)
{
if(b[i]<a[0])
cout<<a[0]<<endl;
else if(b[i]>a[n-1])
cout<<a[n-1]<<endl;
else
{
k=f(a,n,b[i]);
if(abs(b[i]-a[k])<=abs(b[i]-a[k+1]))
cout<<a[k]<<endl;
else
cout<<a[k+1]<<endl;
}
}
return 0;
}
在这个问题中需要注意的有一点,就是你求出小于它的最大的对应的下标之后,还需要和该下标的后面一个数进行比较,看谁更接近。谨记一定要前者小于等于后者,否则就会出错。因为当你如果两者都接近,例如前者为3,后者为5,你所输入的数字为4,判定就是abs(3-4)是否小于abs(5-4),答案是否定的,所以你将输出5,因此就错了。因为题意是:若有多个值满足条件,输出最小的一个。
还需要注意的一点就是,如果你要算的数字小于该数组的第一个数,就可以直接输出该数组的第一个数;
如果你所输入的数字大于该数组的最后一个数字,则输出该数组的最后一个数。