[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=+;
 int n,a[][N],b[N*N];

 void solve()
 {
     scanf("%d",&n);
     for(int i=;i<n;i++) {
         for(int j=;j<;j++) {
             scanf("%d",&a[j][i]);
         }
     }

     for(int i=;i<n;i++) {
         for(int j=;j<n;j++) {
             b[i*n+j]=a[][i]+a[][j];
         }
     }
     sort(b,b+n*n);
     LL res=;
     for(int i=;i<n;i++) {
         for(int j=;j<n;j++) {
             int p=-(a[][i]+a[][j]);
             res+=upper_bound(b,b+n*n,p)-lower_bound(b,b+n*n,p);
         }
     }

     printf("%lld\n",res);

 }
 int main()
 {
     solve();

     return ;
 }