OJ 1794 集合加法

描述

给出2个正整数集合A = {pi | 1 <= i <= a}，B = {qj | 1 <= j <= b}和一个正整数s。问题是：使得pi + qj = 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

分析

p[i]表示在一组数据中，数为i的树有p[i]个，p[sum-i]表示使(i+b=sum)的数b有多少个，相乘即得总数。
减少循环次数就好了。

实现

#include <iostream>
#include <cstring>
using namespace std;
int p[10001];
int q[10001];

int main()
{
//  freopen("in.txt", "r", stdin);
    int cases;
    cin >> cases;
    while (cases--) {
        memset(p, 0, sizeof(p));
        memset(q, 0, sizeof(q));

        int sum, pp, qq, t;
        cin >> sum;
        cin >> pp;
        for (int i = 0; i < pp; i++) {
            cin >> t;
            p[t]++;
        }

        cin >> qq;
        for (int i = 0; i < qq; i++) {
            cin >> t;
            q[t]++;
        }

        int count = 0;
        for (int i = 1; i <= 10000; i++) {
            if (p[i] > 0 && q[sum - i] > 0) {
                count += p[i] * q[sum - i];
            }
        }
        cout << count << endl;
    }
}