PAT (Advanced Level) Practice A1107 Social Clusters （30 分）（C++）(甲级)（并查集）

原题链接：A1107 Social Clusters

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;

const int MAX = 1001;
int i, N, K;
int father[MAX], cnt[MAX] = {0}, hobby[MAX] = {0};
int findFather(int x)
{
    int xFa = x, z = x;//xFa为x的根节点
    while(xFa != father[xFa]) xFa = father[xFa];
    while(x != father[x])//路径压缩；让x~xFa上的所有节点都指向xFa
    {
        z = x;
        x = father[x];
        father[z] = xFa;
    }
    return xFa;
}
void unionSet(int a, int b)//合并ab所在的两个集合
{
    int aFa = findFather(a);
    int bFa = findFather(b);
    if(aFa != bFa) father[bFa] = aFa;
}
bool cmp(int a, int b)//从大到小排序
{
  return a > b;
}

int main()
{
    scanf("%d", &N);
    for(i=0; i<MAX; i++) father[i] = i;//初始化每个节点都是一个根节点
    for(i=1; i<=N; i++)
    {
        scanf("%d: ", &K);
        for(int j=0, x; j<K; j++)
        {
            scanf("%d", &x);
            if(!hobby[x]) hobby[x] = i;//第一个拥有该兴趣的人
            unionSet(i, hobby[x]);//合并此人与第一个拥有该兴趣的人的集合
        }
    }
    for(i=1; i<=N; i++) cnt[findFather(i)]++;//计数每个集合中节点的个数
    sort(cnt, cnt+MAX, cmp);//从大到小排序
    for(i=0; i<MAX && cnt[i]; i++);//集合个数
    printf("%d\n%d", i, cnt[0]);//打印
    for(int j=1; j<i; j++) printf(" %d", cnt[j]);
    return 0;
}