刚开始想的是前先计算前两组的和,另二组排序。然后一组从后往前扫,一组从前往后扫,计算满足和为0的数目。时间复杂度为O(n^3)。不出所料,超时了。
后来各种想,换成分别计算两组和,然后用二分法。这样时间复杂度为O(n^2*lgn)。AC了。
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
int a[4001],b[4001],c[4001],d[4001],x[16000001],y[16000001];
long long ans;
int cmp(const void *a,const void *b)
{
return *(int *)a-*(int *)b;
}
int main()
{
//freopen("t.txt","r",stdin);
int n,i,k1,k2,j;
while(scanf("%d",&n)!=EOF)
{
for(i=0;i<n;i++)
scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
k1=0;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
x[k1++]=a[i]+b[j];
k2=0;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
y[k2++]=-c[i]-d[j];
qsort(y,k2,sizeof(y[0]),cmp);
ans=0;
for(i=0;i<k1;i++)
{
j=lower_bound(y,y+k2,x[i])-y;//可以自己写。。。本来我是自己写的,后来看到了有这样的一个库函数。
for(;j<k2;j++)
if(y[j]==x[i])
ans++;
else break;
}
printf("%lld\n",ans);
}
return 0;
}