L2-005 集合相似度

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

给定两个整数集合，它们的相似度定义为：$N_c/N_t\times 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的size相加得到sum，然后用set_intersection求出交集（两组数中都有的元素）的个数num，然后通过 num/(sum-num)即可得到结果
!!!如果便利两个set的元素加到一个set中，最后一个用例会TLE!!!

题解

#include <bits/stdc++.h>
#define hh ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;

vector<set<int> > v;
int main(){
   hh;
   int n, k, x, a, b;
   cin >> n;
   for(int i = 0; i < n; i++){
       cin >> k;
       set<int> s;
       while(k--){
           cin >> x;
           s.insert(x);
       }
       v.push_back(s);
   }
   cin >> n;
   for(int i = 0; i < n; i++){
       cin >> a >> b;
       int sum = v[a - 1].size() + v[b - 1].size();
       set<int> tmp;
       set_intersection(begin(v[a - 1]), end(v[a - 1]), begin(v[b - 1]), end(v[b - 1]),inserter(tmp, begin(tmp)));
       int num = tmp.size();
       sum -= num;
       cout << setiosflags(ios::fixed) << setprecision(2) << (num * 1.0 / sum * 100) << "%" << endl;
   }
   return 0;
}

提交结果