1107 Social Clusters (30 分)

1107 Social Clusters (30 分)

题目大意

有n个人，每个人喜欢k个活动，如果两个人有任意一个活动相同，就称为他们处于同一个社交网络。求这n个人一共形成了多少个社交网络

基本思路

数据结构：
定义并查集数组father，写好相应的init、findFather、Union函数。
定义数组course，course[i]表示第i门课的课代表是coursr[i]；
定义数组isRoot，下标为用户编号，值为这个社团的用户人数(只有下标为根用户的情况，值才不为0)。
对于每一个人，依次读入它所感兴趣的课程编号。如果该课程没有课代表，就将它自身作为课代表；然后将这个人和该课程课代表所在的集合合并。
统计根用户所在社团的用户人数。
统计跟用户的个数。
输出。

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int MAXN = 1001;
int n;
vector<int> father;
vector<int> isRoot;//下标为某个社团代表人的编号，值为这个社团的人数
int course[MAXN] = {0};//下标为第i个爱好，值为这个爱好的代表人物编号 
int cmp1(int a, int b){
    return a > b;
}
//查找
int findFather(int v){
    if(father[v]==v)
        return v;
    else{
        int F = findFather(father[v]);
        father[v] = F;
        return F;
    }
}
//合并
void Union(int a,int b){
    int faA = findFather(a);
    int faB = findFather(b);
    if(faA!=faB){
        father[faA] = faB;
    }
}
int main(){
    //读入人数，重新定义数组father和isRoot的大小
    cin >> n;
    father.resize(n + 1);
    isRoot.resize(n + 1);
    //初始化并查集
    for (int i = 1; i <= n;i++){
        father[i] = i;
    }
    //遍历1~n个人物
    //读入人物i的爱好数量
    //读入每个爱好的同时，检查这个爱好是否有代表人物
    //如果没有，给这个爱好设置一个代表人物，就设置为当前人物i
    //将当前人物i作为合并函数的一个固定参数，每个爱好的代表人物所在社团的代表人物作为合并另一个参数，调用合并函数
    for (int i = 1; i <= n;i++){
        int k;
        scanf("%d:", &k);
        for (int j = 1; j <= k;j++){
            int ans;
            cin >> ans;
            if(course[ans]==0)
                course[ans] = i;
            Union(i, findFather(course[ans]));
        }
    }
    //统计每个社团代表人所在社团的人数
    //遍历1~n个人物，将isRoot[findFather(i)]++;
    for (int i = 1; i <= n;i++){
        isRoot[findFather(i)]++;
    }
    //统计社团个数
    //遍历1~n个人物，如果isRoot[i]!=0，将cnt++
    int cnt = 0;
    for (int i = 1; i <= n;i++){
        if(isRoot[i]!=0)
            cnt++;
    }
    //将数组isRoot按人数逆序排序
    sort(isRoot.begin(), isRoot.end(),cmp1);
    //输出
    cout<<cnt<<endl;
    for(int i = 0; i < cnt; i++) {
        printf("%d", isRoot[i]);
        if(i != cnt - 1) printf(" ");
    }
    return 0;
}