团体程序设计天梯赛L3-003 社交集群

本文链接：https://blog.youkuaiyun.com/a664607530/article/details/59715661

本文介绍了一种社交网络中寻找兴趣相同用户形成的集群的方法。通过输入用户的兴趣爱好，利用并查集算法找出所有集群，并输出集群数量及各集群规模。

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

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个空格分隔，行首尾不得有多余空格。

输入样例：

输出样例：

3
4 3 1

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <set>
#include <stack>
#include <map>
#include <climits>

using namespace std;

#define LL long long

const int INF=0x3f3f3f3f;

int f[1009];
struct node
{
    int f,sum;
}x[1009],b[1009];

int Find(int k)
{
    if(x[k].f==k) return k;
    else return x[k].f=Find(x[k].f);
}

bool cmp(node x,node y)
{
    return x.sum>y.sum;
}

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int k,a;
        for(int i=1;i<=n;i++) x[i].f=i,x[i].sum=1;
        for(int i=1;i<=1000;i++) f[i]=-1;
        for(int i=1;i<=n;i++)
        {
            scanf("%d:",&k);
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&a);
                if(f[a]==-1) f[a]=i;
                else
                {
                    int xx=Find(f[a]),yy=Find(i);
                    if(xx!=yy)
                    {
                        x[xx].f=yy;
                        x[yy].sum+=x[xx].sum;
                    }
                }
            }
        }
        int cnt=0;
        for(int i=1;i<=n;i++)
            if(x[i].f==i) b[cnt++]=x[i];
        sort(b,b+cnt,cmp);
        printf("%d\n%d",cnt,b[0].sum);
        for(int i=1;i<cnt;i++) printf(" %d",b[i].sum);
        printf("\n");
    }
    return 0;
}