题目详情 - L3-003 社交集群 (30 分) (pintia.cn)
题目是要我们把每个人有部分兴趣相同的集合在一起(不是单纯的求交集,比如有个集合都算是一个集群,因为第一个集合和第二个集合有交集,第二个集合和第三个集合有交集,类比一下函数的传递性就行了),本来想用并查集写的,但并查集不好判断大小。所以只能使用stl大法。
我的思路是将每个人的兴趣都用集合存起来,然后用一个双循环遍历集合,对每对集合求交集,如果存在交集,那就求它们的并集并把并集赋给第二个循环最开始的那个集合的前一个集合,并且将这个集合(不是最开始的那个集合)”去掉“(已经有集合和它匹配了,其他和它匹配的集合照样能够和那个集合匹配,算在一个群体),之后重新开始第二次遍历。
本人语言水平有限,不能把我的意思完全表示出来,只能通过代码描述。。。。。。。。
Code Segment:
#include<cstdio>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
int n,k,h,i,j,a[1001];
set<int> s[1001],ms,ss;
multiset<int,greater<int>> m;
vector<int> v;
scanf("%d",&n);
for(i=0;i<n;i++)
{
scanf("%d:",&k);
for(j=0;j<k;j++)
{
scanf("%d",&h);
s[i].insert(h);//每个个体的兴趣
}
}
for(i=0;i<n;i++)
if(!ms.count(i))
{
ss.insert(i);
for(j=i+1;j<n;j++)
if(!ms.count(j))//”去掉“已经合并的集合,防止重复计算
{
set_intersection(s[i].begin(),s[i].end(),s[j].begin(),s[j].end(),inserter(v,v.begin()));
if(v.size())
{
set_union(s[i].begin(),s[i].end(),s[j].begin(),s[j].end(),inserter(v,v.begin()));
s[i].clear();
ms.insert(j);
for(auto it=v.begin();it!=v.end();it++)
s[i].insert(*it);
ss.insert(j);//存放每个集群的元素
j=i;//从最开始遍历,因为合并后的集群可能和前面的集合有交集
}
v.clear();
}
m.insert(ss.size());
ss.clear();
}
printf("%d\n",m.size());
if(m.size())
{
auto it=m.begin();
printf("%d",*it);
it++;
for(;it!=m.end();it++)
printf(" %d",*it);
printf("\n");
}
}