2020-12-26 社交集群

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：

输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：

其中K_i(>0)是兴趣爱好的个数，h_i[j]是第j个兴趣爱好的编号，为区间 [1, 1000] 内的整数。

输出格式：

首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔，行末不得有多余空格。

输入样例：

8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4

输出样例：

3
4 3 1

题解

并查集的题。设置两个数组，一个用来存为同一个社交群体的用户，一个存爱好。

将爱好数组 hobby 全部设置为 -1

然后每次遇到一个新的爱好，先看这个爱好对应的数值值是不是 -1

• 是 -1：说明这个爱好是第一次出现，之前没有人有过相同的爱好，把 hobby 数组中对应的值改为这个人的编号
• 不是 -1：说明有人有相同的爱好，将这两个人的集合合并

最后排个序就可以了

#include <bits/stdc++.h>
using namespace std;
ostream &sp(ostream &output);
bool cmp(int a, int b)
{
    return a > b;
}
int father[1010];
int hobby[1010];
int findFather(int x)
{
    if (x == father[x])
        return x;
    return findFather(father[x]);
}
int main()
{
    std::ios::sync_with_stdio(false);
    // fill(hobby, hobby + 1010, -1);
    for (int i = 0; i < 1010; ++i)
    {
        father[i] = i;
        hobby[i] = -1;
    }
    int n;
    cin >> n;
    char colon;
    for (int i = 0; i < n; ++i)
    {
        int habitsNum;
        cin >> habitsNum;
        cin >> colon;
        for (int j = 0; j < habitsNum; ++j)
        {
            int hobbyID;
            cin >> hobbyID;
            if (hobby[hobbyID] == -1) //说明还没有人和这个人有一样的爱好
                hobby[hobbyID] = i;
            else
            {
                // 把有相同爱好的人合并到一个并查集中
                int a = findFather(i);
                int b = findFather(hobby[hobbyID]);
                if(a != b)
                {
                    father[a] = b;
                }
            }
        }
    }
    int cnt = 0;
    int ans[1010];
    fill(ans, ans + 1010, 0);
    for (int i = 0; i < n; ++i)
    {
        // cout << "findFather(i) = " << findFather(i) << endl;
        if (i == findFather(i))
            cnt++;
        ans[findFather(i)]++;
    }
    sort(ans, ans + 1010,cmp);// sort(ans,ans+cnt,cmp) 只能过 2 个点
    cout << cnt << endl;
    for (int i = 0; i < cnt - 1; ++i)
        cout << ans[i] << “ ”;
    cout << ans[cnt - 1] << endl;
    return 0;
}