[POJ] 2785 4 Values whose Sum is 0(双向搜索)

题目地址：http://poj.org/problem?id=2785

#include<cstdio>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<stdbool.h>
#include<time.h>
#include<stdlib.h>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
#define clr(x,y)    memset(x,y,sizeof(x))
#define sqr(x)      ((x)*(x))
#define rep(i,a,b)  for(int i=(a);i<=(b);i++)
#define LL          long long
#define INF         0x3f3f3f3f
#define A           first
#define B           second

const int N=4000+131;
int n,a[5][N],b[N*N];

void solve()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++) {
        for(int j=0;j<4;j++) {
            scanf("%d",&a[j][i]);
        }
    }
    
    for(int i=0;i<n;i++) {
        for(int j=0;j<n;j++) {
            b[i*n+j]=a[0][i]+a[1][j];
        }
    }
    sort(b,b+n*n);
    LL res=0;
    for(int i=0;i<n;i++) {
        for(int j=0;j<n;j++) {
            int p=-(a[2][i]+a[3][j]);
            res+=upper_bound(b,b+n*n,p)-lower_bound(b,b+n*n,p);
        }
    }
    
    printf("%lld
",res);
        
}
int main()
{
    solve();
    
    return 0;
}