PAT L2-005 集合相似度 STL-SET

L2-005. 集合相似度

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

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

之后一行给出一个正整数K（<=2000），随后K行，每行对应一对需要计算相似度的集合的编号（集合从1到N编号）。数字间以空格分隔。

输出格式：

对每一对需要计算的集合，在一行中输出它们的相似度，为保留小数点后2位的百分比数字。

输入样例：

33 99 87 1014 87 101 5 877 99 101 18 5 135 18 9921 21 3

输出样例：

50.00%33.33%

我用set做，最后一个测试点超时。

set的用法： 常见用法

                    set的交集，并集

#include <iostream>
#include<set>
#include<cstdio>
#include<algorithm>
using namespace std;

int main()
{
	int a[51][10005];
	int m[51];
	int n,i,q,j,c,b;
	scanf("%d",&n);
	for(i=1;i<=n;i++){
		scanf("%d",&m[i]);
		for(j=1;j<=m[i];j++)scanf("%d",&a[i][j]);
	}
	scanf("%d",&q);
	double result;
	while(q--){
		scanf("%d%d",&c,&b);
		set<int> s1 (a[c]+1,a[c]+m[c]+1);
		set<int> s2 (a[b]+1,a[b]+m[b]+1);
		set<int> s3,s4;
		set_union(s1.begin(),s1.end(),s2.begin(),s2.end(),inserter(s3,s3.begin()));
		set_intersection(s1.begin(),s1.end(),s2.begin(),s2.end(),inserter(s4,s4.begin()));
		result=s4.size()*1.0/s3.size();
		printf("%.2f%\n",result*100);
	}
    return 0;
}

然后我觉得超时的原因是用交集并集对这题来说多余了，因为我们只在意交集，并集的大小，所以超时应该就是求交并太费时间了，于是我就修改了一下，直接比较元素。一开始脑子没转过来，忘记了set中没有重复元素的特性。

#include <iostream>
#include<cstdio>
#include<vector>
#include<set>
using namespace std;

int main()
{
  int n,m,q,temp,i,a,b;
  scanf("%d",&n);
  vector<set<int> >v(n);
  for(i=0;i<n;i++){
    scanf("%d",&m);
    set<int > s;
    while(m--){
      scanf("%d",&temp);
      s.insert(temp);
    }
    v[i]=s;
  }
  scanf("%d",&q);
  int bing,jiao;
  while(q--){
    scanf("%d%d",&a,&b);
    a--,b--;
    jiao=0,bing=v[a].size();
    set<int>::iterator it;
    for(it=v[b].begin();it!=v[b].end();it++){
      if(v[a].find(*it)!=v[a].end())jiao++;
      else bing++;
    }
    printf("%.2f%\n",jiao*1.0/bing*100);

  }
    return 0;
}