题意:给出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] 有多少个
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include <cmath> 5 #include<stack> 6 #include<vector> 7 #include<map> 8 #include<set> 9 #include<queue> 10 #include<algorithm> 11 #define mod=1e9+7; 12 using namespace std; 13 14 typedef long long LL; 15 const int maxn=4000+5; 16 int s[maxn*maxn]; 17 int a[maxn],b[maxn],c[maxn],d[maxn]; 18 19 int main(){ 20 int t,n,i,j,cnt; 21 scanf("%d",&t); 22 while(t--){ 23 scanf("%d",&n); 24 for(i=0;i<n;i++) scanf("%d %d %d %d",&a[i],&b[i],&c[i],&d[i]); 25 26 int len=0; 27 for(i=0;i<n;i++){ 28 for(j=0;j<n;j++) 29 s[len++]=a[i]+b[j]; 30 } 31 32 sort(s,s+len); 33 34 cnt=0; 35 for(i=0;i<n;i++){ 36 for(j=0;j<n;j++) 37 cnt += upper_bound(s,s+len, -c[i]-d[j]) - lower_bound(s,s+len, -c[i]-d[j]); 38 } 39 cout<<cnt<<" "; 40 if(t) cout<<" "; 41 } 42 }