百练2792:集合加法

2792:集合加法

• 查看
• 提交
• 统计
• 提示
• 提问

总时间限制: 

3000ms 

内存限制: 

65536kB

描述

给出2个正整数集合A = {p_i | 1 <= i <= a}，B = {q_j | 1 <= j <= b}和一个正整数s。问题是：使得p_i + q_j= s的不同的(i, j)对有多少个。

输入

第1行是测试数据的组数n，后面跟着n组测试数据。

每组测试数据占5行，第1行是和s (1 <= s <= 10000)，第2行是一个正整数a (1 <= a <= 10000)，表示A中元素的数目。第3行是a个正整数，每个正整数不超过10000，表示A中的元素。第4行是一个正整数b (1 <= b <= 10000)，表示B中元素的数目。第5行是b个正整数，每个正整数不超过10000，表示B中的元素。

注意：这里的集合和数学书上定义的集合有一点点区别——集合内可能包含相等的正整数。

输出

n行，每行输出对应一个输入。输出应是一个非负整数。

样例输入

299249 49250 501191 2 3 4 5 6 7 8 91010 9 8 7 6 5 4 3 2 1

样例输出

49

• 查看 
• 提交 
• 统计 
• 提示 
• 提问
• 代码1：快排+二分法

  #include<cstdio>#include<algorithm>#include<cstring>using namespace std;const int maxn = 10002;int A[maxn],B[maxn];int main(){int n,a,b,s;scanf("%d",&n);while(n--){memset(A,0,sizeof(A));memset(B,0,sizeof(B));scanf("%d%d",&s,&a);for(int i = 0;i<a;++i)scanf("%d",&A[i]);scanf("%d",&b);for(int i=0;i<b;i++)scanf("%d",&B[i]);int ans = 0,len = b,pos;sort(A,A+a);sort(B,B+b);for(int i = 0;i<a;i++){if(A[i]>=s) break;pos = lower_bound(B,B+len,s-A[i]) - B;if(pos < len && B[pos] == s-A[i]){while(B[pos++] == s-A[i]) ans++;}len = pos;}printf("%d\n",ans);}return 0;}

  代码2：哈希，桶排序

  #include<cstdio>#include<cstring>using namespace std;const int maxn = 10002;int A[maxn],B[maxn],C[maxn];int main(){int n,a,b,s,tmp,k,ans;scanf("%d",&n);while(n--){memset(A,0,sizeof(A));memset(B,0,sizeof(B));scanf("%d%d",&s,&a);k = ans = 0;for(int i = 0;i<a;++i){scanf("%d",&tmp);if(A[tmp] == 0){C[k++]=tmp;A[tmp]++;}elseA[tmp]++;}scanf("%d",&b);for(int i =0;i<b;i++){scanf("%d",&tmp);B[tmp]++;}for(int i =0;i<k;i++){if(s-C[i]>0){ans += A[C[i]]*B[s-C[i]];}}printf("%d\n",ans);}return 0;}