【PTA 天梯赛】L3-003 社交集群(并查集)

当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式:

输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:

K​i​​: h​i​​[1] h​i​​[2] ... h​i​​[K​i​​]

其中K​i​​(>0)是兴趣爱好的个数,h​i​​[j]是第j个兴趣爱好的编号,为区间 [1, 1000] 内的整数。

输出格式:

首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔,行末不得有多余空格。

输入样例:

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 p[],ai[],cir[];
int find(int r)
{
if(p[r]!=r)
p[r]=find(p[r]);
return p[r];
}
void join(int x,int y)
{
int fx=find(x),fy=find(y);
if(fx!=fy)
p[fx]=fy;
}
int main()
{
int n,i,j,k,x;
char ch;
cin>>n;
for(i=;i<=;i++)
p[i]=i; for(i=;i<=n;i++)
{
cin>>k>>ch;
for(j=;j<=k;j++)
{
cin>>x;
if(ai[x]==)ai[x]=i;//i号人作为爱好x的父亲
join(i,find(ai[x]));//将i号人与爱好x的人群连接
}
}
int sum=;
for(i=;i<=n;i++)
cir[find(i)]++;
sort(cir+,cir+,greater<int>());//按从大到小排序
for(i=;i<=;i++)
{
if(cir[i]==)
{
i--;
break;
}
}
cout<<i<<endl;
for(j=;j<=i;j++)
{
if(j==)cout<<cir[j];
else cout<<" "<<cir[j];
}
cout<<endl;
return ;
}
上一篇:php部分---面向对象,设计模式(单例模式、工厂模式)、oop六大原则;


下一篇:flutter圆角效果的实现