7-29 集合相似度
给定两个整数集合,它们的相似度定义为:Nc/Nt×100%。其中Nc是两个集合都有的不相等整数的个数,Nt是两个集合一共有的不相等整数的个数。你的任务就是计算任意一对给定集合的相似度。
输入格式:
输入第一行给出一个正整数N(≤50),是集合的个数。随后N行,每行对应一个集合。每个集合首先给出一个正整数M(≤104),是集合中元素的个数;然后跟M个[0,109]区间内的整数。
之后一行给出一个正整数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合并成一个set然后就可以直接知道有多少公共元素。
各种优化但是都逃不过超时。
#include<iostream>
#include<unordered_set>
#include<vector>
using namespace std;
int main(){
int n,a,b;
scanf("%d",&n);
vector<unordered_set<int>>ss(n+1);
for(int i=1;i<=n;i++){
int t;
scanf("%d",&t);
int p;
for(int l=0;l<t;l++){
scanf("%d",&p);
ss[i].insert(p);
}
}
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d %d",&a,&b);
unordered_set<int>t;
for(auto it=ss[a].begin();it!=ss[a].end();it++)t.insert(*it);
for(auto it=ss[b].begin();it!=ss[b].end();it++)t.insert(*it);
printf("%0.2f%%\n",100.0*(ss[a].size()+ss[b].size()-t.size())/t.size());
}
return 0;
}
换成 set内置的方法 .cout()返回出现次数因为set是无重复的所以 1和0分别表示存在与否。
#include<iostream>
#include<unordered_set>
#include<vector>
using namespace std;
int main(){
int n,a,b;
scanf("%d",&n);
vector<unordered_set<int>>ss(n+1);
for(int i=1;i<=n;i++){
int t;
scanf("%d",&t);
int p;
for(int l=0;l<t;l++){
scanf("%d",&p);
ss[i].insert(p);
}
}
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d %d",&a,&b);
unordered_set<int>t;
int count=0;
for(auto it=ss[a].begin();it!=ss[a].end();it++) if(ss[b].count(*it))count++;
printf("%0.2f%%\n",100.0*count/(ss[a].size()+ss[b].size()-count));
}
return 0;
}