L3-003 社交集群(并查集)

题目链接

思路：
并查集 + unordered_set
先初始并查集f数组，再输入将每人的爱好id加入到各自的set集合中，
从前往后遍历每个人为 i，再遍历 i 后面的人 j，若有与 i 相同的兴趣爱好的 j 则合并sz(维护并查集大小)和set(并查集所含爱好id)，然后break。
这里有个小优化：当某个元素 j 已经被前面的某个 i 合并过后，就可以不用再当 i 了，因为已经跟着前面的 i 去过后面一次了。
当然这里是维护并查集大小，所以不能重复加入并查集，要判断f1 != f2，这里wa了一发。

代码：

#include <bits/stdc++.h>
#define fastio ios::sync_with_stdio(false), cin.tie(NULL), cout.tie(NULL)
#define debug(a) cout << "debug : " << (#a) << " = " << a << endl

using namespace std;

typedef long long ll;
typedef pair<int, string> PII;

const int N = 1e3 + 10;
const int INF = 0x3f3f3f3f;
const double eps = 1e-6;
const int mod = 998244353;

unordered_set<int> a[N];
int f[N], sz[N];
bool vis[N], vis2[N];
vector<int> res;

int find(int x)
{
    if (f[x] != x)
        f[x] = find(f[x]);
    return f[x];
}

int main()
{
    for (int i = 1; i <= 1000; i++)
        f[i] = i, sz[i] = 1;
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        char c;
        int k, x;
        cin >> k >> c;
        while (k--)
        {
            cin >> x;
            a[i].insert(x);
        }
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = i + 1; j <= n; j++)
        {
            for (auto p : a[j])
            {
                if (a[i].find(p) != a[i].end())
                {
                    int f1 = find(i), f2 = find(j);
                    if (f1 != f2)
                    {
                        f[f2] = f1;
                        for (auto p : a[f2])
                            a[f1].insert(p);
                        sz[f1] += sz[f2];
                        vis[j] = true;
                    }
                    break;
                }
            }
        }
    }
    for (int i = 1; i <= n; i++)
    {
        int fa = find(i);
        if (!vis[fa])
        {
            vis[fa] = true;
            res.push_back(sz[fa]);
        }
    }
    sort(res.begin(), res.end());
    int Size = res.size();
    cout << Size << endl;
    for (int i = Size - 1; i >= 0; i--)
    {
        if (i != Size - 1)
            cout << ' ';
        cout << res[i];
    }
    return 0;
}