PAT L2-005. 集合相似度

 

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

 

 一开始做法是，手动在每输入完一个数组后对数组进行去重2333，找相同的值是O(n2)的时间复杂度，前四个样例还是能过的，第五个应该是数据量太大吧， TLE了。后来在网上看到了STL有set这个容器，才了解了这个题要考的东西..

 

关于set的用法，链接如下：点击打开链接

附上代码：

#include<stdio.h>
#include<set>
#include <iostream>
using namespace std;

set<int> u[55];

void find(int a, int b)
{
  int same = 0;
  set<int>::iterator it;
  for( it=u[a].begin(); it!=u[a].end(); it++ ){
    if( u[b].find(*it) != u[b].end() ){
      same++;
    }
  }
  int nt = u[a].size() + u[b].size() - same;
  double res = same*1.0/nt*100.0;
  printf("%.2lf%%\n", res);

}
int main()
{
  int n;
  cin >>n;
  for( int i=1; i<=n; i++ ){
    int a, data;
    cin >>a;
    for( int j=0; j<a; j++ ){
      cin >>data;
      u[i].insert(data);
    }
  }
  int m;
  cin >>m;
  for( int i=0; i<m; i++ ){
    int c, d;
    cin >>c >>d;
    find(c, d);
  }

  return 0;
}