A1107 Social Clusters(并查集)

题目描述

题意：有N个人，每个人喜欢若干项活动，如果两个人有任意一个活动相同，那么就称他们处于同一个社交网络，求这N个人总共形成了多少个社交网络

非常经典的并查集问题，有点小变化的是统计集合个数和每个集合有多少人，只要开个数组统计以某节点为根的集合有多少个即可，代码中使用isRoot数组实现

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

int n;
int father[1005];
int hobby[1005]={0};
int isRoot[1005]={0};

bool cmp(int x,int y){
    return x>y;
}

int findFather(int x)
{
    if(father[x]==-1) return x;
    else{
        int tmp=findFather(father[x]);
        father[x]=tmp;
        return tmp;
    }
}

void Union(int a,int b)
{
    int ta=findFather(a);
    int tb=findFather(b);
    if(ta!=tb)
        father[ta]=tb;
}


int main()
{
    memset(father,-1,sizeof(father));
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int k;
        scanf("%d%*c",&k);
        for(int j=0;j<k;j++){
            int t;
            scanf("%d",&t);
            if(hobby[t]==0)
                hobby[t]=i;
            else
                Union(i,findFather(hobby[t]));
        }
    }
    for(int i=1;i<=n;i++)
        isRoot[findFather(i)]++;
    int ans=0;
    for(int i=1;i<=n;i++){
        if(isRoot[i]!=0)
            ans++;
    }
    printf("%d\n",ans);
    sort(isRoot+1,isRoot+n+1,cmp);
    for(int i=1;i<=ans;i++){
        printf("%d",isRoot[i]);
        if(i<ans) printf(" ");
    }
}