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行，每行输出对应一个输入。输出应是一个非负整数。
样例输入
2
99
2
49 49
2
50 50
11
9
1 2 3 4 5 6 7 8 9
10
10 9 8 7 6 5 4 3 2 1
样例输出
4
9
查看 提交 统计 提示 提问
代码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]++;
}else
A[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;
}
