链接: http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1710
Description
有A、 B、 C 三个集合的,其中a∈A, b ∈ B, c ∈ C,求有多少种方式使得a + b = c。
Input
有多组测试数据,请处理到文件结束。
对于每组测试数据,有三行:
第一行为A集合的描述,第一个数为n,表示A集合有n个数,接下来有n个整数a1~an。
第二行为B集合,第三行为C集合,表示含义参考第一行。
每个集合中的数两两不相等。
1<=n<=5000,|ai| <= 1 000 000 000
Output
对于每组测试数据,输出一行,包含一个整数,表示有多少种组合方式。
Sample Input
3 -10 4 -6
3 -10 3 -1
3 3 -4 -6
3 -8 -9 -4
3 9 -10 2
3 -8 -7 5
3 -4 -6 -2
3 9 -9 -2
3 3 -13 -4
Sample Output
2
2
3
代码如下:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <map>
#include <vector>
#include <algorithm>
#define MAXN 5005
#define RST(N)memset(N, 0, sizeof(N))
using namespace std;
int a[MAXN], b[MAXN], c[MAXN];
int a_n, b_n, c_n, res;
int cmp(const void *a, const void *b)
{
return *(int *)a - *(int *)b;
}
void Init()
{
RST(a), RST(b), RST(c), res = 0;
for(int i=0; i<a_n; i++) scanf("%d", &a[i]);
scanf("%d", &b_n);
for(int i=0; i<b_n; i++) scanf("%d", &b[i]);
scanf("%d", &c_n);
for(int i=0; i<c_n; i++) scanf("%d", &c[i]);
qsort(a, a_n, sizeof(a[0]), cmp);
qsort(b, b_n, sizeof(b[0]), cmp);
}
void solve()
{
for(int i=0; i<c_n; ++i) {
for(int j=0, k=b_n-1; j<a_n && k>=0; ) {
if(a[j] + b[k]> c[i]) k--;
else if(a[j] + b[k]< c[i]) j++;
else j++, k--, res++;
}
}
printf("%d\n", res);
}
int main()
{
while(~scanf("%d", &a_n)) {
Init(), solve();
}
return 0;
}