把a+b与c+d的结果储存到两个数组里面,再排序,“快排”一样从头到尾和从尾到头遍历。注意输出不要PE;
#include<bits/stdc++.h>
using namespace std;
//-1 -1 1 1
int main(void){
int n;
cin>>n;
while(n--){
int l;
scanf("%d",&l);
int a[l],b[l],c[l],d[l];
for(int i=0;i<l;i++){
scanf("%d %d %d %d",&a[i],&b[i],&c[i],&d[i]);
}
int ll=l*l;
int ab[ll],cd[ll];
for(int i=0;i<l;i++){
for(int j=0;j<l;j++){
ab[i*l+j]=a[i]+b[j];
cd[i*l+j]=c[i]+d[j];
}
}
sort(ab,ab+ll);
sort(cd,cd+ll);
// for(int i=0;i<ll;i++){
// printf("%d ",ab[i]);
// }
// for(int i=0;i<ll;i++){
// printf("%d ",cd[i]);
// }
int ct=0;
int p=0,q=ll-1;
while(p<ll&&q>-1){
if(ab[p]+cd[q]==0){
int t1=p,t2=q;
while(1){
t1++;
// printf("%d\n",t1);
if(t1==ll||ab[t1]!=ab[p]){
break;
}
}
// printf("abt1=%d,abp=%d\n",ab[t1],ab[p]);
while(t2--){
if(t2==-1||cd[t2]!=cd[q]){
break;
}
}
// printf("t1=%d,t2=%d,ll=%d\n",t1,t2,ll);
ct+=(t1-p)*(q-t2);
p=t1;
q=t2;
}
else if(ab[p]+cd[q]>0){
q--;
}
else {
p++;
}
// printf("ct=%d\n",ct);
}
cout<<ct<<endl;
if(n){
printf("\n");
}
}
}