题解:如果从4个数列中选择的话总共有n^4种情况,所以全都判断一遍不可行。不过将它们对半分成AB和CD考虑,就可以快速解决了,从2个数列中选择的话只有n^2种组合,所以可以进行枚举。先从A,B种取出a,b后,为了使总和为0则需要从C,D中取出c+d=-a-b。因此先将A,B中取出a,b后,为了使得总和为0则需要从C,D中取出c+d=-a-b。因此先将C,D中取数字的n^2种方法全都枚举出来,将这些和排好序,这样就可以运用二分搜索了,算法的复杂度是O(n^2logn)。
问题的规模比较大,无法枚举所有元素的组合,但能够枚举一半元素的组合,此时,将问题拆成两半后分别枚举,再合并它们的结果往往比较有效。
附上代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=4e3+50;
int a[maxn],b[maxn],c[maxn],d[maxn];
int cd[maxn*maxn];
int n;
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
cd[i*n+j]=c[i]+d[j];
}
}
sort(cd,cd+n*n);
long long res=0;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
int temp=-(a[i]+b[j]);
res+=upper_bound(cd,cd+n*n,temp)-lower_bound(cd,cd+n*n,temp);
}
}
printf("%lld\n",res);
return 0;
}