编程题#1:集合加法
来源: POJ (http://bailian.openjudge.cn/practice/2792/)
注意: 总时间限制: 3000ms 内存限制: 65536kB
描述
给出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
程序解答方法一:
本程序主要在于用map容器对集合中的重复元素进行了统计,并按元素由小到大进行了排序。内存:1664kB,时间:1476ms
#include <iostream>
#include <map>
#include <algorithm>
using namespace std;
int main() {
int n, s, ni, nj;
int a[10000], b[10000];
int nCount;
map<int, int> seti, setj;
map<int, int>::iterator nii, njj;
cin >> n;
while (n--){
cin >> s;
cin >> ni;
for (int i = 0; i < ni; i++)
cin >> a[i];
cin >> nj;
for (int j = 0; j < nj; j++)
cin >> b[j];
sort(a, a + ni); //按元素由小到大进行排序
sort(b, b + nj);
int temp = -1;
for (int i = 0; i < ni; i++){
if (temp == a[i]) //对集合中的重复元素进行统计
seti[a[i]]++;
else
seti[a[i]] = 1;
temp = a[i];
}
temp = -1;
for (int j = 0; j < nj; j++){
if (temp == b[j])
setj[b[j]]++;
else
setj[b[j]] = 1;
temp = b[j];
}
nCount = 0;
for (nii = seti.begin(); nii != seti.end(); nii++){
for (njj = setj.begin(); njj != setj.end(); njj++){
if (nii->first + njj->first == s)
nCount += nii->second * njj->second;
}
}
cout << nCount << endl;
seti.clear();
setj.clear();
}
return 0;
}
程序解答方法二:
直接暴力求解,内存:1024kB,时间:627ms,我就呵呵了。。。。。。
#include<iostream>
using namespace std;
int sum, numA, numB;
int arrayA[10001], arrayB[10001];
int countHelper(int sum, int arrayA[], int numA, int arrayB[], int numB)
{
int count = 0;
for (int i = 0; i < numA; ++i) {
for (int j = 0; j < numB; ++j) {
if (arrayA[i] + arrayB[j] == sum) {
count++;
}
}
}
return count;
}
int main()
{
int num;
cin>>num;
while(num--)
{
cin>> sum >> numA;
for (int i = 0; i < numA; ++i) {
cin>>arrayA[i];
}
cin>>numB;
for (int j = 0; j < numB; ++j) {
cin>>arrayB[j];
}
int count = countHelper(sum, arrayA, numA, arrayB, numB);
cout<<count<<endl;
}
return 0;
}