STL之二分查找

STL之二分查找

2017.07.20 12:25:21字数141阅读159

需要包含头文件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;
    }

}