L3-003. 社交集群并查集

最新推荐文章于 2024-02-01 17:46:04 发布

西瓜不甜不甜不甜

最新推荐文章于 2024-02-01 17:46:04 发布

阅读量216

点赞数

CC 4.0 BY-SA版权

分类专栏：并查集

本文链接：https://blog.youkuaiyun.com/zqf3535/article/details/79697437

并查集专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个基于共同兴趣进行人群匹配的算法实现。通过读取每个人的兴趣列表，并利用并查集数据结构来判断拥有相同兴趣的人是否已经连接在一起。最终统计并输出各兴趣群组的人数。

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

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

#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn=10005;

int n,k;
int fa[maxn];
int num[maxn];
int val[maxn];

int find(int x){
    if (x==fa[x]) return x;
    return fa[x]=find(fa[x]);
}

void unite(int x,int y){
    x=find(x);
    y=find(y);
    if (x!=y){
        fa[y]=x;
    }
}

int main(){

    cin >> n;
    for (int i=1;i<=n;i++){
        fa[i]=i;
    }

    for (int i=1;i<=n;i++){
        scanf("%d:",&k);
        for (int j=0;j<k;j++){
            int x;
            cin >> x;
            if (!val[x]){      //兴趣x尚未有人喜欢 
                val[x]=i;       
            }
            unite(find(val[x]),i);  //通过相同兴趣，将人与人连接 
        }
    }

    for (int i=1;i<=n;i++){
        num[find(i)]++;
    }
    int cnt=0;
    int ans[maxn];
    for (int i=1;i<=n;i++){
        if (num[i]){
            ans[cnt++]=num[i];
        }
    }

    printf("%d\n",cnt);
    sort(ans,ans+cnt);
    for(int i=cnt-1;i>=0;i--){
        if (i!=cnt-1) cout << " ";
        cout << ans[i];
    }
}