PTA团体程序设计天梯赛-练习集 L3-003 社交集群（并查集）

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。
输入格式：
输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：
Ki: hi[1] hi[2] … hi[Ki]
其中Ki（>0）是第i个人的兴趣的数量，hi[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
思路：将每个人的爱好用book数组来表示，假设一个人的一个爱好为i（这个人的编号为x），如果book【i】为0，则表示在这人之前，还没有人的某个爱好为i，则让book【i】=x，如果book【i】！=0，则使用并查集的join方法合并x和book【i】，使他们属于同一个社交关系圈。最后我们只需要遍历一下，就知道有多少个社交关系圈，以及各个关系圈的人数。

#include <bits/stdc++.h>
using namespace std;
int n,par[1010],book[1010],rel[1010];
void init(int n){
    for(int i = 1;i <= n;i++)
        par[i] = i;
}
int find(int x){
    if(x == par[x]) return x;
    return par[x] = find(par[x]);
}
void join(int x,int y){
    x = find(x);
    y = find(y);
    if(x == y) return;
    else
        par[x] = y;
}
int main(){
    cin>>n;
    init(n);
    int res = 0;
    for(int i = 1;i <= n;i++){
        int k = 0;
        scanf("%d:",&k);
        for(int j = 0;j < k;j++){
            int num = 0;
            cin>>num;
            if(book[num] == 0)
                book[num] = i;
            else
                join(i,book[num]);
        }
    }
    for(int i = 1;i <= n;i++){
    	//这是一个新的社交关系圈
        if(rel[find(i)] == 0) res++;
        //不管是新的还是旧的关系圈，都得往这个关系圈里面加一个人
        rel[find(i)]++;
    }
    cout<<res<<endl;
    sort(rel,rel+1010,[](int a,int b){return a > b;});
    for(int i = 0;i < res;i++){
        if(i != res-1) cout<<rel[i]<<" ";
        else cout<<rel[i];
    }
    return 0;
}