HLG 1710 给出三个集合a,b,c,统计集合a元素+集合b中元素=集合c中的元素的个数 （基础题）

链接： 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;}