L3-003 社交集群(较简单并查集)

最新推荐文章于 2022-04-19 11:14:48 发布

Helium_wild

最新推荐文章于 2022-04-19 11:14:48 发布

阅读量176

点赞数

分类专栏： PAT

本文链接：https://blog.csdn.net/qq_42576687/article/details/88907307

版权

PAT 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

在这里插入图片描述输入样例：
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 <cstdio>
#include <algorithm>
#define sd(a) scanf("%d",&a)
using namespace std;
const int N=2000+20,INF=0x3f3f3f3f;
int f[N],cnt[N],ans[N];
int seek(int x)
{
    if(x==f[x])
        return x;
    return f[x]=seek(f[x]);
}
int main()
{
    int n;sd(n);
    for(int i=1;i<=n;i++)
        f[i]=i,cnt[i]=1;
    for(int i=n+1;i<=1000+n+1;i++)
        f[i]=i,cnt[i]=0;
    for(int i=1;i<=n;i++)
    {
        int k,x;
        scanf("%d:",&k);
        while(k--)
        {
            sd(x),x+=n;
            int ri=seek(i),rx=seek(x);
            if(ri!=rx)
            {
                if(ri<rx)
                    f[rx]=ri,cnt[ri]+=cnt[rx];
                else
                    f[ri]=rx,cnt[rx]+=cnt[ri];
            }
        }
    }
    int tot=0;
    for(int i=1;i<=n;i++)
        if(f[i]==i)
            ans[++tot]=cnt[i];
    sort(ans+1,ans+tot+1);
    printf("%d\n",tot);
    for(int i=tot;i;i--)
        printf("%d%c",ans[i],(i==1)?'\n':' ');
}