PAT 社交集群（并查集）

社交集群

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

输入格式：

输入在第一行给出一个正整数 N（ $\leq 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] 内的整数。

输出格式：

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

输入样例：

输出样例：

3
4 3 1

变相的并查集，根据一样的兴趣找朋友，以前做的都是直接找到后面的，仔细思考一下其实还是比较容易的，附上

AC代码：

#include<iostream>
#include<algorithm>
#define MAX 1010
using namespace std;
int f[MAX],sum[MAX],vis[MAX];
void init()
{
    for(int i = 0;i < MAX; i++)
        f[i] = i;
}
int getf(int v)
{
    if(f[v] == v)
        return v;
    else
    {
        f[v] = getf(f[v]);
        return f[v];
    }
}
void marge(int v,int u)
{
    int a = getf(v);
    int b = getf(u);
    if(a != b)
            f[a] = b;
}
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    int n,d,t;
    char ch;
    cin >> n;
    init();
    for(int i = 1;i <= n; i++)
    {
        cin >> d >> ch;
        for(int j = 0;j < d; j++)
        {
            cin >> t;
            if(!vis[t]) vis[t]=i;
            else    marge(i,vis[t]);
        }
    }
    int x,k = 0,flag = 0;
    for(int i = 1;i <= n; i++)
    {
        x = getf(i);
        if(sum[x] == 0)
            k++;
        sum[x]++;
    }
    cout << k << endl;
    sort(sum,sum+1000,cmp);
    for(int i = 0;i < k; i++)
    {
        if(!flag)
        {
            cout << sum[i];
            flag++;
        }
        else
            cout <<" "<< sum[i];
    }
    return 0;
}