PAT 天梯训练 7-9 集合相似度

PAT 天梯训练 7-9 集合相似度

题目连接
7-9 集合相似度（25 分）
给定两个整数集合，它们的相似度定义为：
N​c/Nt×100%。其中Nc是两个集合都有的不相等整数的个数，Nt是两个集合一共有的不相等整数的个数。你的任务就是计算任意一对给定集合的相似度。
输入格式：

输入第一行给出一个正整数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%

解题思路：
Nc：两个集合都有的不相等整数的个数，列如样例1中，Nc=2(87,101)
Nt：Nt是两个集合一共有的不相等整数个数，列如样例1中，Nt=4(87,101,5,99)
显然采用set集合，Nc为两个集合中共有的元素个数，Nt为两个集合中元素之和减去重复的个数Nc

#include<iostream>
#include<set>
#include<string>
#include<cstring>
#include<cstdio>
#include<iterator>
using namespace std;
set<int>a[55];
int b[55];
int main(){
    int n;
    cin>>n;
    memset(b,0,sizeof(b));
    for(int i=1;i<=n;i++){
        cin>>b[i];
        int t;
        for(int j=0;j<b[i];j++){
            cin>>t;
            a[i].insert(t);
        }
    }
    int m;
    cin>>m;
    int x,y;
    int num;
    for(int i=0;i<m;i++){
        cin>>x>>y;
        num=0;
        set<int>::iterator it=a[x].begin();
        for(;it!=a[x].end();it++){
            if(a[y].find(*it)!=a[y].end()){
                num++;
            }
        }
        double ans=(double)num/(double)((a[x].size()-num)+a[y].size());
        printf("%.2lf%%\n",ans*100.0);
    }
    return 0;
}