PAT-4C-L3-003-社交集群（并查集）

链接：https://www.patest.cn/contests/gplt/L3-003

L3-003. 社交集群

时间限制

1000 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

83: 2 7 101: 42: 5 31: 41: 31: 44: 6 8 1 51: 4

输出样例：

34 3 1

---

提交代码

#include <bits/stdc++.h>using namespace std;int Scan(){    int res=0,ch,flag=0;    if((ch=getchar())=='-')flag=1;    else if(ch>='0'&&ch<='9')res=ch-'0';    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-'0';    return flag?-res:res;}void Out(int a){    if(a>9)Out(a/10);    putchar(a%10+'0');}#define INF 0x3f3f3f3f#define bug cout<<"bug"<<endlconst int MAXN = 1e4+7;int set_[MAXN];int num[MAXN];int set_find(int p){    if(set_[p]==-1)return p;    return set_[p]=set_find(set_[p]);}void make_friend(int a, int b){    a=set_find(a);    b=set_find(b);    if(a!=b)    {        set_[b]=a;        num[a]+=num[b];        num[b]=0;    }}int ans[MAXN];int main(){    int n,m,p;    memset(set_,-1,sizeof(set_));    n=Scan();    for(int i=1; i<=n; ++i)    {        num[1000+i]=1;        m=Scan();        for(int j=0; j<m; ++j)        {            p=Scan();            make_friend(1000+i,p);        }    }    int temp=0;    for(int i=0; i<=2000; ++i)        if(num[i])ans[temp++]=num[i];    Out(temp),putchar('\n');    sort(ans,ans+temp);    for(int i=temp-1; i>0; --i)        Out(ans[i]),putchar(' ');    Out(ans[0]),putchar('\n');    return 0;}