L3-003 社交集群 (30分)

我的思路其实很简单
这题是比较标准的 并查集 类型的题目。
主要思想就是一个常规的并查集数组 f[MAXN] 保存各个成员之间的朋友关系

那么不同兴趣圈的人怎么建立朋友关系呢？
我的办法是通过另一个变量存储不同的兴趣圈里的人是哪些，然后在读取数据的时候，一个人若是发现有兴趣A,则遍历一遍兴趣A所在的兴趣圈， 与圈中的所有人建立一遍朋友关系，为了减少建立朋友关系的次数，这边我选用set来进行保存。

代码如下：

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

set<int> s[10001];
int f[10001];
void init() {
    for(int i = 1; i <= 10001; i++)
        f[i] = i;
}
int find(int x) {
    if(f[x] == x)	return x;
    else {
        int  fx = find(f[x]);
        f[x] = fx;
        return fx;
    }
}
void merge(int a, int b) {
    int fa = find(a), fb = find(b);
    if(fa != fb) {
        f[fa] = fb;
    }
}
bool cmp(int a, int b) {
    return a > b;
}
int main()
{
    int n;
    cin >> n;
    init();
    int k, temp;
    for(int i = 1; i <= n; i++) {
        scanf("%d:", &k);
        for(int j = 0; j < k; j++) {
            cin >> temp;
            for(auto item : s[temp]) {
                merge(i, item);
            }
            s[temp].insert(i);
        }
    }
    int Hash[1001] = {0}, cnt = 0;
    for(int i = 1; i <= n; i++) {
        int fi = find(i);
        Hash[fi]++;
        if(Hash[fi] == 1) cnt++;
    }
    sort(Hash, Hash + 1001, cmp);
    cout << cnt << endl << Hash[0];
    for(int i = 1; i < cnt; i++)
        printf(" %d", Hash[i]);
    return 0;
}