4 Values whose Sum is 0 POJ - 2785(折半枚举(双向搜索))

传送门

题解：如果从4个数列中选择的话总共有n^4种情况，所以全都判断一遍不可行。不过将它们对半分成AB和CD考虑，就可以快速解决了，从2个数列中选择的话只有n^2种组合，所以可以进行枚举。先从A,B种取出a，b后，为了使总和为0则需要从C,D中取出c+d=-a-b。因此先将A,B中取出a，b后，为了使得总和为0则需要从C,D中取出c+d=-a-b。因此先将C,D中取数字的n^2种方法全都枚举出来，将这些和排好序，这样就可以运用二分搜索了，算法的复杂度是O(n^2logn)。

问题的规模比较大，无法枚举所有元素的组合，但能够枚举一半元素的组合，此时，将问题拆成两半后分别枚举，再合并它们的结果往往比较有效。

附上代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

const int maxn=4e3+50;

int a[maxn],b[maxn],c[maxn],d[maxn];
int cd[maxn*maxn];
int n;

int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cd[i*n+j]=c[i]+d[j];
        }
    }
    sort(cd,cd+n*n);
    long long res=0;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            int temp=-(a[i]+b[j]);
            res+=upper_bound(cd,cd+n*n,temp)-lower_bound(cd,cd+n*n,temp);
        }
    }
    printf("%lld\n",res);
    return 0;
}