天梯赛 L3-003. 社交集群

并查集，以人为节点，用set来存每个人的兴趣，输入第i个人的兴趣时，如果在前i个人的兴趣里找得到，就将两个人合并。

#include<iostream>
#include<string>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
#include<queue>
#include<map>
#include<vector>
#define MAX_N 1005
#define ll long long

using namespace std;

int n;
int root[MAX_N];
set<int> ss[MAX_N];
int ans[MAX_N];
void init()
{
    for(int i = 0; i < MAX_N; i++)
        root[i] = i,ans[i] = 1;
}
int findFather(int x)
{
    if(root[x]==x)
        return x;
    return findFather(root[x]);
}

void merge(int x, int y)
{
    int xx = findFather(x);
    int yy = findFather(y);
    
    if(xx!=yy)
    {
        root[yy] = xx;
        ans[xx]+=ans[yy];
        ans[yy] = 0;
    }
}
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    init();
    scanf("%d",&n);
    int k,temp;
    for(int i = 1; i <= n; i++)
    {
        scanf("%d:",&k);
        for(int j = 1; j <= k; j++)
        {
            scanf("%d",&temp);
            ss[i].insert(temp);
            for(int k = 1; k < i; k++)
            {
                if(ss[k].find(temp) != ss[k].end())
                {
                    merge(i,k);
                }
            }
        }
    }
    sort(ans+1,ans+n+1,cmp);
    
    int con = 0;
    for(int i = 1; i <= n; i++)
    {
        if(ans[i])
            con++;
        else
            break;
    }
    printf("%d
",con);
    for(int i = 1; i < con; i++)
        printf("%d ",ans[i]);
    printf("%d
",ans[con]);
    return 0;
}