需要包含头文件algorithm
C++ STL实现二分查找的函数有三个:lower_bound、upper_bound 、binary_search.
他们所有的前提是要有序。
ForwardIter lower_bound(ForwardIter first ,ForwardIter last ,const T& val)
返回[first , last)中第一个>=val的元素;
ForwardIter upper_bound(ForwardIter first ,ForwardIter last ,const T& val)
返回[first , last)中第一个>val的元素;
例题:T^T OJ PID:2311
题目大意:给出两个数组,累计第二个数组中的第一个数组的值。
分析:由于给的范围太大,普通遍历会超时,所以使用二分法。
附上已AC代码
#include<bits/stdc++.h>
using namespace std;
int a[1000005],b[1000005];
int main()
{
int t;
cin>>t;
while(t--)
{
long long int ans=0;
long long int n,q;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
cin>>n>>q;
for(long long int i=1;i<=n;i++)
cin>>a[i];
for(long long int i=1;i<=q;i++)
{
cin>>b[i];
if(b[i]==a[lower_bound(a+1,a+n+1,b[i])-a])
ans+=b[i];
}
cout<<ans<<endl;
}
}