L2-005 集合相似度 (25 分)
给定两个整数集合,它们的相似度定义为:N
c
/N
t
×100%。其中N
c
是两个集合都有的不相等整数的个数,N
t
是两个集合一共有的不相等整数的个数。你的任务就是计算任意一对给定集合的相似度。
输入格式:
输入第一行给出一个正整数N(≤50),是集合的个数。随后N行,每行对应一个集合。每个集合首先给出一个正整数M(≤10
4
),是集合中元素的个数;然后跟M个[0,10
9
]区间内的整数。
之后一行给出一个正整数K(≤2000),随后K行,每行对应一对需要计算相似度的集合的编号(集合从1到N编号)。数字间以空格分隔。
输出格式:
对每一对需要计算的集合,在一行中输出它们的相似度,为保留小数点后2位的百分比数字。
输入样例:
3
3 99 87 101
4 87 101 5 87
7 99 101 18 5 135 18 99
2
1 2
1 3
输出样例:
50.00%
33.33%
我一开始是用set加数组的形式存的,这没问题,但是我后来查的时候又开了一个set,然后遍历两遍,太慢了。
如下:
#include<iostream>
#include<iomanip>
#include<set>
#include<vector>
using namespace std;
set<int>ch[101];
double Check(int x,int y){set<int>zan;for (set<int>::iterator it = ch[x].begin();it!=ch[x].end();it++){zan.insert(*it);}for (set<int>::iterator it = ch[y].begin();it!=ch[y].end();it++){zan.insert(*it);}double zong = ch[x].size()+ch[y].size();double nc = zong - zan.size();double nt = zan.size();double sum = 100.0*nc/nt;return sum;
}
int main()
{int n;cin>>n;for (int i=1;i<=n;i++){int k,num;cin>>k;while (k--){cin>>num;ch[i].insert(num);}}cin>>n;int a,b;for (int i=0;i<n;i++){cin>>a>>b;double cha = Check(a,b);cout<<fixed<<setprecision(2)<<cha<<"%"<<endl;}return 0;
}
所以我后来改进了一下:
#include<iostream>
#include<iomanip>
#include<set>
#include<vector>
using namespace std;
set<int>ch[101];
double Check(int x,int y){int cou=0;for (set<int>::iterator it = ch[x].begin();it!=ch[x].end();it++){if (ch[y].find(*it)!=ch[y].end()) cou++; }int nt=ch[x].size()+ch[y].size()-cou;return 1.0*cou/nt*100;
}
int main()
{int n;cin>>n;for (int i=1;i<=n;i++){int k,num;cin>>k;while (k--){cin>>num;ch[i].insert(num);}}cin>>n;int a,b;for (int i=0;i<n;i++){cin>>a>>b;double cha = Check(a,b);cout<<fixed<<setprecision(2)<<cha<<"%"<<endl;}return 0;
}
过喽!芜湖!
STL库,yyds。