POJ 2785 4 Values whose Sum is 0

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<algorithm>
using namespace std;

const int maxn=4000+10;
int n,tot;
long long a[maxn][6];
long long Q[20000000];
long long P[20000000];
struct X
{
long long val;
long long num;
}s[20000000];

long long work(long long tag)
{
int l=0,r=tot;
while(l<=r)
{
int mid=(l+r)/2;
if(s[mid].val<tag) l=mid+1;
else if(s[mid].val==tag) return s[mid].num;
else r=mid-1;
}
return 0;
}

int main()
{
while(~scanf("%d",&n))
{
for(int i=1; i<=n; i++)
for(int j=1; j<=4; j++)
scanf("%lld",&a[i][j]);

int cnt=0;
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
Q[cnt++]=a[i][1]+a[j][2];
sort(Q,Q+cnt);

tot=0;
s[0].val=Q[0]; s[0].num=1;
for(int i=1;i<cnt;i++)
{
if(Q[i]==Q[i-1]) s[tot].num++;
else
{
tot++;
s[tot].num=1;
s[tot].val=Q[i];
}
}
cnt=0;
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
P[cnt++]=a[i][3]+a[j][4];

long long ans=0;
for(int i=0;i<cnt;i++) ans=ans+work(-P[i]);
printf("%lld\n",ans);
}
return 0;
}