poj 2785 4 Values whose Sum is 0 (二分+sort）

题意：每一列中选一个数使得相加之和为0，问有几组这样的四个数。

先把两列中选出的数相加，然后再二分查找满足条件的（可以算是分治思想)。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

int a[4010],b[4010],c[4010],d[4010];
int sum1[4010*4010],sum2[4010*4010];
int r,cnt;

void find_(int x)
{
    int low=0,high=r-1,mid;
    while(low<high)
    {
        mid = (low+high)/2;
        if(sum2[mid]<x)
            low=mid+1;
        else
            high=mid;
    }
    while(sum2[low]==x && low<r)
    {
        cnt++;
        low++;
    }
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
            scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
        r=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
                sum1[r++]=a[i]+b[j];
        }
        r=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
                sum2[r++]=c[i]+d[j];
        }
        cnt=0;
        sort(sum2,sum2+r);
        for(int i=0;i<r;i++)
            find_(-sum1[i]);
        printf("%d\n",cnt);
    }
    return 0;
}