PTA L2-005 集合相似度 C++实现 详细解析

给定两个整数集合，它们的相似度定义为：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%

解题思路：

理解题意后，可以提炼出相似度的计算方法为： 找出去重后两集合的相同元素个数 cnt，再用 (cnt / (去重后集合A的大小 + 去重后集合B的大小 - cnt) ) * 100% 

Tips：set 容器具有去重功能，其 count 方法是数某个元素在 set 容器中有几个。

代码示例：

#include <bits/stdc++.h>

using namespace std;

struct stu
{
    set<int> num;
};

stu res[100]; //定义结构体数组res 其内部含set容器

int main()
{
    int n;
    cin >> n;

    for (int i = 1; i <= n; i++) //从1开始编号
    {
        int m;
        cin >> m;

        while (m--)
        {
            int temp;
            cin >> temp;
            res[i].num.insert(temp);
        }
    }

    int k;
    cin >> k;

    while (k--)
    {
        int index1, index2;
        cin >> index1 >> index2;

        double cnt = 0;
        
        //这里的for循环里面 index1与index2可以交换位置 因为其逻辑是一样的 不影响最终结果
        for (auto p = res[index1].num.begin(); p != res[index1].num.end(); p++)
        {
            //set含有去重功能 因此count函数只有可能返回0或1
            if (res[index2].num.count(*p)) 
            {
                cnt++;
            }
        }
        
        //计算相似度
        int sum_size = res[index1].num.size() + res[index2].num.size();
        double ans = (cnt / (sum_size - cnt)) * 100;

        printf("%.2f", ans);
        cout << "%" << endl;
    }

    return 0;
}

运行结果：