PTA 社交集群

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：
输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：

K​i​​ : h​i​​ [1] h​i​​ [2] … h​i​​ [K​i​​ ]其中K​i (>0)是兴趣爱好的个数，h​i​​ [j]是第j个兴趣爱好的编号，为区间 [1, 1000] 内的整数。

输出格式：
首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔，行末不得有多余空格。

输入样例：
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<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n;
int f[1010];
void init()
{
    for(int i=0; i<1010; i++)
        f[i]=i;
}
int getf(int v)
{
    if(f[v]==v)
        return v;
    else
        return f[v]=getf(f[v]);
}
void merge(int v,int u)
{
    int t1,t2;
    t1=getf(v);
    t2=getf(u);
    if(t1!=t2)
        f[t2]=t1;
}
bool cmp(int x,int y)
{
    return x>y;
}
int book[1010],num[1010];
int main()
{
    scanf("%d",&n);
    init();
    int a;
    for(int i=0; i<n; i++)
    {
        int k;
        scanf("%d: ",&k);
        scanf("%d",&a);
        book[i]=a;
        for(int j=1; j<k; j++)
        {
            scanf("%d",&a);
            merge(book[i],a);
        }
    }
    for(int i=0; i<n; i++)
    {
        int x=getf(book[i]);
        num[x]++;
    }
    int s=0;
    for(int i=0; i<1010; i++)
    {
        if(num[i])
            s++;
    }
    sort(num,num+1010,cmp);
    printf("%d\n",s);
    for(int i=0; i<s; i++)
    {
        if(i==0)
            printf("%d",num[i]);
        else
            printf(" %d",num[i]);
    }
    return 0;
}