题意:给出n,四个集合a,b,c,d每个集合分别有n个数,分别从a,b,c,d中选取一个数相加,问使得a+b+c+d=0的选法有多少种
看的紫书,先试着用hash写了一下,
是用hash[]记录下来a[i]+b[j]的值,
如果a[i]+b[j]>0,hash[a[i]+b[j]]=1
如果a[i]+b[j]<0,hash[-(a[i]+b[j])]=-1
再用一个hash0[]去储存c[i]+d[j] 这样只需要满足hash[i]==1||hash0[i]==-1或者hash[i]==-1,hash0[i]==1就可以了
可是这样写过不了样例,因为在选择的过程中,不一定只有一种选择的和为i,最后可能会将本应该符合题意的hash值覆盖掉
然后学习的lrj的代码,记录下所有a[]+b[]的值,在这个序列里面找-c[i]-d[j] 有多少个
#include<iostream>
#include<cstdio>
#include<cstring>
#include <cmath>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<algorithm>
#define mod=1e9+7;
using namespace std; typedef long long LL;
const int maxn=+;
int s[maxn*maxn];
int a[maxn],b[maxn],c[maxn],d[maxn]; int main(){
int t,n,i,j,cnt;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(i=;i<n;i++) scanf("%d %d %d %d",&a[i],&b[i],&c[i],&d[i]); int len=;
for(i=;i<n;i++){
for(j=;j<n;j++)
s[len++]=a[i]+b[j];
} sort(s,s+len); cnt=;
for(i=;i<n;i++){
for(j=;j<n;j++)
cnt += upper_bound(s,s+len, -c[i]-d[j]) - lower_bound(s,s+len, -c[i]-d[j]);
}
cout<<cnt<<"\n";
if(t) cout<<"\n";
}
}