【折半枚举 && 二分】POJ - 2785 4 Values whose Sum is 0

Problem Description

给定各有n个整数的四个数列A, B, C, D。要从每个数列中取出1个数，使四个数的和为0。求出这样的组合的个数。当一个数列中有多个相同的数字时，把他们作为不同的数字看待。

思路:

正常思路全枚举 O(n^4)肯定超时。不过将它们对半分成AB和CD再考虑，就可以快速解决了。
对于C,D枚举所有和的情况O(n^2), 对于A,B枚举所有和的情况O(n^2)。
c+d = -(a+b)。
我们可以将C,D所有和先存数组里，然后枚举A,B所有和的负数 在数组中寻找这个数的个数。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 4001;
int a[maxn], b[maxn], c[maxn], d[maxn];
int cd[maxn*maxn];
int main()
{
    int n, i, j;
    while(~scanf("%d", &n))
    {
        for(i = 0; i < n; i++)
        {
            for(j = 0; j < 4; j++)
            {
                if(j == 0) scanf("%d", &a[i]);
                if(j == 1) scanf("%d", &b[i]);
                if(j == 2) scanf("%d", &c[i]);
                if(j == 3) scanf("%d", &d[i]);
            }
        }
        int cnt = 0;
        for(i = 0; i < n; i++)
        {
            for(j = 0; j < n; j++)
                cd[cnt++] = c[i] + d[j];
        }
        sort(cd, cd+cnt);
        int ans = 0;
        for(i = 0; i < n; i++)
        {
            for(j = 0; j < n; j++)
            {
                int t = -(a[i]+b[j]);
                //lower_bound返回第一个出现的值大于等于t的位置
                //upper_bound返回最后一个大于等于t的位置
                ans += upper_bound(cd, cd+cnt, t) - lower_bound(cd, cd+cnt, t);
            }
        }
        printf("%d\n", ans);
    }
}