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,0,sizeof(int) * s);
        memset(arr_b,0,sizeof(int) * s);

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

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

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

        cout << count << endl;

        delete [] arr_a;
        delete [] arr_b;
    }

    return 0;
}