UVa 1152 4 Values whose Sum is 0

题意:给出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 }
View Code
原文地址:https://www.cnblogs.com/wuyuewoniu/p/4367570.html