4 Values whose Sum is 0 (二分)

 

 思路：分治法

 枚举4组数O（n^4）绝对超时

 因此可以使用分治的思想，将 4 组数分为两组，然后每组再分别计算和，最后对两组合进行排序，让正数与负数相加判断是否为 0 即可

 

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

using namespace std;

int a[4001];
int b[4001];
int c[4001];
int d[4001]; 
int n;
int s1[20000000];
int s2[20000000];

int main()
{
    while(scanf("%d", &n) != EOF)
    {
        for(int i = 1; i <=n; ++i)
            scanf("%d %d %d %d", &a[i], &b[i], &c[i], &d[i]);
        
        int k = 1;
        for(int i = 1; i <= n; ++i)
        {
            for(int j = 1; j <= n; ++j)
            {
                s1[k] = a[i] + b[j];
                s2[k] = c[i] + d[j];
                ++k;
            }
        }
        
        sort(s1+1, s1+k);
        sort(s2+1, s2+k);
        
        int right = k - 1;
        int ans = 0;
        for(int left = 1; left < k; ++left)
        {
            while(right > 0 && s1[left] + s2[right] > 0)
                right--;
                
            if(right < 0)
                break;
                
            int tmp = right;    // 一组left可能对应多组right解 
            while(right > 0 && s1[left] + s2[right] == 0)
            {
                ans++;
                right--;
            }    
            right = tmp;        // 下一组left可能和当前这一组left相同，所以rigiht要复原 
        }
        
        printf("%d\n", ans);
    }
    

    return 0;
}