1107. Social Clusters

最新推荐文章于 2022-11-10 19:05:39 发布

原创最新推荐文章于 2022-11-10 19:05:39 发布 · 196 阅读

0 ·

CC 4.0 BY-SA版权

PAT甲级专栏收录该内容

43 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决具有共同爱好人群分类的问题。通过构建并查集实现快速查找与合并操作，有效组织数据，最终统计并输出各群体规模。代码实现了输入处理、并查集初始化、成员合并及查找等功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

有共同爱好的属于一个集合
并查集！
注意可以先合并再查每个人属于哪个集合

#include <bits/stdc++.h>
using namespace std;
#define FORP(i,a,b) for(int i=a;i<=b;i++)
set<int> s[1025];
int fa[1024];
int cnt[1024];
int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
void Union(int x,int y){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy) fa[fy]=fx;
}
int main(){
    FORP(i,1,1023){
        fa[i]=i;
    }
    int n;cin>>n;
    FORP(i,1,n){
        int k;
        cin>>k;
        getchar();
        int rt;
        cin>>rt;
        s[i].insert(rt);
        FORP(j,2,k){
            int data;
            cin>>data;
            s[i].insert(data);
            Union(rt,data);
        }
    }
    FORP(i,1,n){
        cnt[find(*s[i].begin())]++;
    }
    vector<int> ans;
    FORP(i,1,1023){
        if(cnt[i]){
            ans.push_back(-cnt[i]);
        }
    }
    sort(ans.begin(),ans.end());
    cout<<ans.size()<<endl;
    FORP(i,0,ans.size()-1){
        printf("%d%c",-ans[i],i==ans.size()-1?'\n':' ');
    }
    return 0;   
}