UVA - 1152 4 Values whose Sum is 0 【中途相遇法/二分】

题目链接：点击打开链接

题意：

给你4个数组，每个数组有n个元素。从每个数组里各取一个数，使其和为0，求有多少组这样的a,b,c,d。

思路：

可以将4个数组分为两部分，sum1[ i ]=A[ i ]+B[ i ]，sum2[ i ]=C[ i ]+D[ i ]。那么只需要求有多少个sum1[ i ]+sum2[ j ]==0就可以了，即sum1[ i ]==sum2[ j ]。其实在这里我们可以省略sum2数组。直接查找会超时。这里用到了lower_bounder和upper_bounder函数。

ForwardIter lower_bound(ForwardIter first, ForwardIter last,const _Tp& val)算法返回一个非递减序列[first, last)中的第一个大于等于值val的位置。

ForwardIter upper_bound(ForwardIter first, ForwardIter last, const _Tp& val)算法返回一个非递减序列[first, last)中第一个大于val的位置。

那么 upper_bound(ForwardIter first, ForwardIter last, const _Tp& val) - lower_bound(ForwardIter first, ForwardIter last,const _Tp& val) 就会返回数组中val值的个数。算法要求数组有序，内部用二分查找，效率非常快。

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define MAXN 4005
int n;
long long A[MAXN],B[MAXN],C[MAXN],D[MAXN],sum[MAXN*MAXN];

int main()
{
    int t,flag=0;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=0;i<n;i++)
            scanf("%lld%lld%lld%lld",&A[i],&B[i],&C[i],&D[i]);
        int cnt=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                sum[cnt++]=A[i]+B[j];
        sort(sum,sum+cnt);
        int ans=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                ans+=upper_bound(sum,sum+cnt,-C[i]-D[j])-lower_bound(sum,sum+cnt,-C[i]-D[j]);
            }
        }
        if(flag) printf("\n");
        printf("%d\n",ans);
        flag=1;
    }
    return 0;
}