团体程序设计天梯赛-练习集 L3-003 社交集群 并查集 解题报告

Posted 2022-06-22 小夥

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

输入样例：

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 <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
#define MAXN 1010
int c[MAXN], f[MAXN];

int find(int x)

    return x == f[x] ? x : f[x] = find(f[x]);


void merge(int x, int y)

    x = find(x);
    y = find(y);
    if(x != y) f[x] = y;


int main()

    int n, v, t, tt;
    scanf("%d", &n);
    for(int i = 0; i < 1010; i++) f[i] = i;
    for(int i = 0; i < n; i++)
    
        scanf("%d: ", &t);
        scanf("%d", &tt);
        c[i] = tt;
        for(int j = 1; j < t; j++)
        
            scanf("%d", &v);
            merge(tt, v);
        
    
    int num[1010] = 0, cnt = 0;
    for(int i = 0; i < n; i++)
        num[find(c[i])] ++;
    for(int i = 0; i < 1010; i++)
        if(num[i]) cnt++;
    sort(num, num + 1010, greater<int>());
    printf("%d\\n", cnt);
    printf("%d", num[0]);
    for(int i = 1; i < cnt; i++)
        printf(" %d", num[i]);
    printf("\\n");
    return 0;