在社交网络平台注册时,用户通常会输入自己的兴趣爱好,以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序,找出所有的集群。
输入格式:
输入的第一行给出正整数N(<=1000),即社交网络中的用户总数(则用户从1到N编号)。随后N行,每行按下列格式列出每个人的兴趣爱好:
Ki: hi[1] hi[2] ... hi[Ki]
其中Ki(>0)是第i个人的兴趣的数量,hi[j]是第i个人的第j项兴趣的编号,编号范围为[1, 1000]内的整数。
输出格式:
首先在第一行输出整个网络中集群的数量,然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例: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;
const int MAXN=1e3+7;
const int inf=1e9;
int n,m,k;
vector<int>head[MAXN];
int pre[MAXN];
int vis[MAXN];
int ans[MAXN];
int findx(int x)
{
return pre[x]==x?x:pre[x]=findx(pre[x]);
}
void he(int x,int y)
{
int fx=findx(x);
int fy=findx(y);
if(fx!=fy)pre[fy]=fx;
}
bool cmp(int x,int y)
{
return x>y;
}
int main()
{
scanf("%d",&n);
for(int i=0; i<n; ++i)
{
scanf("%d:",&m);
int x;
while(m--)
{
scanf("%d",&x);
head[x].push_back(i);
}
}
for(int i=0; i<n; ++i)
{
pre[i]=i;
}
for(int i=1; i<=1000; ++i)
{
int l=head[i].size();
if(l>=2)
{
int x=head[i][0];
for(int j=1; j<l; ++j)
{
int y=head[i][j];
he(x,y);
}
}
}
//求出圈子的个数
int cnt=0;
for(int i=0;i<n;++i)if(pre[i]==i)cnt++;
printf("%d\n",cnt);
//统计每个圈子的人数
for(int i=0;i<n;++i)
{
ans[findx(i)]++;
}
//排序,输出前cnt个
sort(ans,ans+n,cmp);
for(int i=0;i<cnt;++i)
{
if(i)printf(" ");
printf("%d",ans[i]);
}
return 0;
}