L3-1 社交集群 (30分) -2018年天梯赛模拟题集

L3-1 社交集群 (30分)

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

输入格式：

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

K**i: h**i[1] h**i[2] … h**i[K**i]

其中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

~

并查集问题。类似部落那道题，不过部落那道题是给的每一行是属于一个集合的，直接在那一行里做merge即可，而这个给的是，每一个人在哪些兴趣小组里。把每个小组里的人存到一起，再对每个小组里的人进行merge。

对了，这个是部落那个题。7-5 部落 (20分) HBU-DS_Chantec的博客-CSDN博客

这个是 朋友圈那个题[2-4 朋友圈 (25分)]

对了，这个题集里面，我感觉也是并查集的题，感觉比这个题还难

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-D13KYQHI-1606568882296)(C:\Users\梁子超\AppData\Roaming\Typora\typora-user-images\image-20201128205846681.png)]

代码

#include <iostream>
#include <algorithm>
#include <map>
#include <vector>
using namespace std;
int fa[1001];
void init(int);
int find(int x);
void merge(int x, int y);
int main()
{
    int N;
    cin >> N;
    init(N);
    int cnt = 1;
    vector<vector<int>> v;
    map<int, int> idx;
    for (int i = 1; i <= N; ++i)
    {
        //每一行
        int k;
        scanf(" %d:", &k);
        for (int c = 0; c < k; ++c)
        {
            int cur;
            scanf(" %d", &cur);
            int tmpidx = idx[cur];
            if (tmpidx != 0)
                v[tmpidx - 1].push_back(i);
            else
            {
                idx[cur] = cnt++;
                vector<int> tmp;
                tmp.push_back(i);
                v.push_back(tmp);
            }
        }
    }

    for (int i = 0; i < v.size(); ++i)
    {
        for (int k = 1; k < v[i].size(); ++k)
            merge(v[i][0], v[i][k]);
    }
    int cnt2 = 0;
    for (int i = 1; i <= N; ++i)
    {
        if (i == find(i))
            cnt2++;
    }
    cout << cnt2 << endl;

    vector<int> cnt_(N + 1);
    for (int i = 1; i <= N; ++i)
    {
        cnt_[find(i)]++;
    }
    sort(cnt_.rbegin(), cnt_.rend());
    for (auto it = cnt_.begin(); it != cnt_.end(); ++it)
    {
        if (*it == 0)
            break;
        if (it != cnt_.begin())
            cout << " ";
        cout << *it;
    }
}

void init(int N)
{
    for (int i = 0; i <= N; ++i)
        fa[i] = i;
}
int find(int i)
{
    if (fa[i] == i)
        return i;
    else
        return fa[i] = find(fa[i]);
}
void merge(int x, int y)
{
    fa[find(y)] = find(x);
}