PAT(天梯赛练习题) -L2-005. 集合相似度（stl）

题目链接

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

输入格式：

输入第一行给出一个正整数N（<=50），是集合的个数。随后N行，每行对应一个集合。每个集合首先给出一个正整数M（<=10⁴），是集合中元素的个数；然后跟M个[0, 10⁹]区间内的整数。

之后一行给出一个正整数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%

集合题目，注意坑点是如果用stl的话一定不能对两个集合合并，因为会卡爆。

（还有集合不是不能有相同元素的吗）。

思想是用stl的set把数据存起来（顺便去重），然后通过对这询问的set做交集计算（就是统计相同的有多少），一定不能合并set。

代码如下

#include<iostream>
#include<string>
#include<cstring>
#include<vector>
#include<map>
#include<algorithm>
#include<queue>
#include<set>
#include<cstdio>
#include<functional>
#include<iomanip>
#include<cmath>
#include<stack>
using namespace std;
const int maxn = int(1e4) + 100;
const int inf = 0x7fffffff;
typedef long long LL;
typedef unsigned long long ull;
vector<set<int> >ve;
int main() {
  //freopen("D:\\cpp\\test.txt", "r", stdin);
  int n, m, k, q, x, y;
  scanf("%d", &n);
  for (int i = 0; i < n; i++) {
    scanf("%d", &m);
    set<int> tmp;
    for (int j = 0; j < m; j++) {
      scanf("%d", &k);
      tmp.insert(k);
    }
    ve.push_back(tmp);
  }
  scanf("%d", &q);
  while (q--) {
    scanf("%d%d", &x, &y);
    x--, y--;
    int sim = 0, tot = ve[y].size();
    for (set<int>::iterator it = ve[x].begin(); it != ve[x].end(); ++it) {
      if (!ve[y].count(*it)) {
        tot++;
      }
      else {
        sim++;
      }
    }
    double ans = double(sim) / double(tot);
    printf("%.2lf%%\n", ans*100.0);
  }
}