1063 Set Similarity

1063 Set Similarity_第1张图片

题目大意:

计算两个集合的相似度,相似度定义为(两集合交集元素数量)/(两集合并集元素数量)。

解题思路:

就是STL set函数的运用,如果采用循环得到并集最后一个样例会超时。这里要用到set_union()函数,得到的结果保存在数组中而不是集合。至于交集元素的数量可以通过集合合并前后元素数量相减得到。
代码如下:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int main()
{
//  std::ios::sync_with_stdio(false);
//  cin.tie(0);
//  freopen("test.txt","r",stdin);
//  freopen("output.txt","w",stdout);
 set<int> s[60];
 int n;
 scanf("%d",&n);
 for(int i=1;i<=n;i++)
 {
   int t;
   scanf("%d",&t);
   while(t--)
   {
     int k;
     scanf("%d",&k);
     s[i].insert(k);
   }
 }
 int q;
 scanf("%d",&q);
 while(q--)
 {
   int a,b;
   scanf("%d %d",&a,&b);
   int len=s[a].size()+s[b].size();
   vector<int> tmp;//保存并集
   //set::iterator it;
   set_union(s[a].begin(), s[a].end(), s[b].begin(), s[b].end(), back_inserter(tmp));//两集合的并集
   //for(it=s[a].begin();it!=s[a].end();it++)tmp.insert(*it);
   //for(it=s[b].begin();it!=s[b].end();it++)tmp.insert(*it);
   double nc=len-tmp.size();//交集元素的数量
   //cout<
   double sim=nc/tmp.size()*100;//计算相似度
   printf("%.1f%c\n",sim,'%');
 }
 return 0;
}

你可能感兴趣的:(STL,PAT甲级)