L2-005 集合相似度 (25 分)

L2-005 集合相似度 (25 分)

题目

给定两个整数集合，它们的相似度定义为：N**c/N**t×100%。其中N**c是两个集合都有的不相等整数的个数，N**t是两个集合一共有的不相等整数的个数。你的任务就是计算任意一对给定集合的相似度。

输入格式：

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

思路

首先对样例的第一个询问做一个解释：

相同的数字是87，101；不同的数字是99，87，101，5.所以是50.00%。

我的思路是用set对所有集合进行存储+去重，不用vector的原因是set在之后每次的询问中还可以起到一个查找的作用。

具体见代码

代码

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int N = 1e4 + 19;
const ll mod = 1e9 + 7;

set<int> st[N];

double calc(int x, int y)
{
    int cnt = 0;
    int n = st[x].size();
    int m = st[y].size();
    for(auto i : st[y])
    {
        if(st[x].find(i) != st[x].end())
        {
            n--;
            cnt++;
        }
    }
    return 100.0 * (double)cnt / (double)(n + m);
}

int main()
{
    int n, k;
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        int m;
        cin >> m;
        while(m--)
        {
            int x;
            cin >> x;
            st[i].insert(x);
        }
    }
    cin >> k;
    while(k--)
    {
        int x, y;
        cin >> x >> y;
        printf("%.2f\%\n", calc(x, y));
    }
    return 0;
}