这是一道很典型的并查集问题
每一个人用第一个兴趣编号作为代表,将每个人的兴趣编号用并查集合并起来,这样有相同兴趣的人就被分在了一个组,然后遍历每一个人,查找他的代表兴趣的根节点,然后用一个数组算出每个圈子有多少人,用set集合统计圈子个数。
写完运行时,输出的第二行总是一堆很长很长的数字,百度之后发现是因为排序函数的输入变量和整体的变量重名了,改一下就好啦~~~但其实浪费了很多时间。。。
原题:
当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入样例:
8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4
输出样例:
3
4 3 1
#include<bits/stdc++.h>
using namespace std;
int n,k,x,t;
int fa[1100];
int flag[1100];
int hobby[1100];
int Find(int a)//找到祖先结点
{
if(fa[a]==a)
return a;
else return Find(fa[a]);
}
void Merge(int a,int b)//合并
{
int w1 = Find(a);
int w2 = Find(b);
if(w1!=w2)
fa[w2] = w1;//注意这里谁是谁的父节点
}
bool cmp(int s1,int s2)
{
return s1>s2;
}
int main()
{
cin>>n;
for(int i = 1;i<=1100;i++)
{
fa[i] = i;
}
char c;
for(int i = 1;i<=n;i++)
{
cin>>k>>c;
cin>>x;//第一个兴趣作为代表
flag[i] = x;
k--;
while(k>0)
{
cin>>t;
Merge(x,t);
k--;
}
}
set<int> s;
memset(hobby,0,sizeof(hobby));
for(int i = 1;i<=n;i++)
{
int w = Find(flag[i]);//注意这里是用函数找根节点,而不是找父节点
s.insert(w);
hobby[w]++;
}
cout<<s.size()<<endl;
sort(hobby,hobby+1001,cmp);
for(int i = 0;i<s.size();i++)
{
if(i)
cout<<" ";
printf("%d",hobby[i]);
}
}