OpenJudge 2792 集合加法

1.链接地址：

http://bailian.openjudge.cn/practice/2792

2.题目：

总Time Limit:

3000ms

Memory Limit:

65536kB

Description

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

Input

第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中的元素。

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

Output

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

Sample Input

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

Sample Output

4
9

3.思路：

标记法，记录每个数字出现的次数，再计算

4.代码：

 #include <iostream>
 #include <cstdio>
 #include <cstring>

 using namespace std;

 int main()
 {
     //freopen("C://input.txt","r",stdin);

     int temp,i;

     int n;
     cin >> n;

     while(n--)
     {
         int s,a,b;

         cin >> s;

         int *arr_a = new int[s];
         int *arr_b = new int[s];

         memset(arr_a,,sizeof(int) * s);
         memset(arr_b,,sizeof(int) * s);

         cin >> a;
         while(a--)
         {
             cin >> temp;
             if(temp <= s) ++(arr_a[temp - ]);
         }

         cin >> b;
         while(b--)
         {
             cin >> temp;
             if(temp <= s) ++(arr_b[temp - ]);
         }

         int count = ;
         for(i = ; i < s; ++i)
         {
             count += arr_a[i] * arr_b[s - i - ];
         }

         cout << count << endl;

         delete [] arr_a;
         delete [] arr_b;
     }

     return ;
 }