POJ 2785_4 Values whose Sum is 0

题意：

A,B,C,D四组数中各取一个数，使这四个数相加为0，问有多少组取法？

分析：

四个数列有$n^4$种取法，规模较大，但是可以将他们分成AB,CD两组，分别枚举，最后再合并。

代码：

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn = 4005, maxm = 16000025;
int  cd[maxm], a[maxn], b[maxn],c[maxn],d[maxn];
int main (void)
{
    int n;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];
        }
    }

    ll res = 0;
    sort(cd, cd+n*n);
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            int tmp = -(a[i] + b[j]);
            res += upper_bound(cd, cd + n * n, tmp) - lower_bound(cd, cd + n * n, tmp);
        }
    }
    printf("%I64d\n",res);
}