POJ2785 4 Values whose Sum is 0（折半枚举+二分）

题目大意

给定A B C D四个数组，每个数组里面有n（n<=40000）个数

问从四个数组里各选一个数，相加等于零的情况有多少种

（如果两个数值相同而位置不同算同一种情况）

思路

如果是四重循环找，不用说绝对超时

那么就需要用到一些优化了

其实我们可以先将C和D的情况先枚举出来（两重循环）并对其排序

然后再用两重循环将A和B的和的情况取相反数

再用upper_bound和lower_bound找到在有序数组内有没有相等的数，累加答案

（upper_bound找到大于的数，lower_bound找到大于等于的数，相减就是结果）

#include<cstdio>#include<cstring>#include<algorithm>#define ll long longusing namespace std;ll A[4005],B[4005],C[4005],D[4005];ll fcd[16000005],ans,s;int n,len;int main(){    ans=0;len=0;    scanf("%d",&n);    for(int i=1;i<=n;++i)    scanf("%lld%lld%lld%lld",&A[i],&B[i],&C[i],&D[i]);    for(int i=1;i<=n;++i)     for(int j=1;j<=n;++j)      fcd[++len]=C[i]+D[j];//枚举C和D的所有方案    sort(fcd+1,fcd+len+1);//lower_bound和upper_bound一定要有序    for(int i=1;i<=n;++i)     for(int j=1;j<=n;++j)     {         s=-(A[i]+B[j]);//记得是早相反数         ans+=upper_bound(fcd+1,fcd+len+1,s)-lower_bound(fcd+1,fcd+len+1,s);     }    printf("%lld\n",ans);    return 0;}