【GPLT 三阶题目集】L3-003 社交集群

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

输入格式：

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

输出格式：

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

输入样例：

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

//只要两人爱好列表中有一个相同，则他们属于同一个社交集群
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;

const int MAX = 1e3 + 10;
int anc[MAX];
int n;
vector<vector<int>> like;

int findd(int x)
{
    if (anc[x] == x)
        return x;
    return anc[x] = findd(anc[x]);
}

void unionn(int a, int b)
{
    int a_anc = findd(a);
    int b_anc = findd(b);
    if (a_anc != b_anc)
        anc[a_anc] = b_anc;
    return;
}

bool isSimilar(int a, int b) //判断两人爱好是否有相同点
{
    for (int i : like[a])
        for (int j : like[b])
            if (i == j)
                return true;
    return false;
}

bool cmp(int a, int b)
{
    return a > b;
}

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i++)
        anc[i] = i;
    like.resize(n + 1);
    for (int i = 1; i <= n; i++) //记录每个人的爱好
    {
        vector<int> tem;
        int k; cin >> k; getchar();
        while (k--)
        {
            int x; cin >> x;
            tem.push_back(x);
        }
        like[i] = tem;
    }
    for (int i = 1; i <= n; i++) //爱好相似的两人合并
        for (int j = i + 1; j <= n; j++)
            if (isSimilar(i, j))
                unionn(i, j);
    set<int> res; //社交集群，统计个数
    int cot[MAX] = { 0 }; //社交集群人数
    for (int i = 1; i <= n; i++)
    {
        int x = findd(i);
        res.insert(x);
        cot[x]++;
    }
    cout << res.size() << endl;
    sort(cot + 1, cot + n + 1, cmp); //有效编号1~n
    for (int i = 1; i <= res.size(); i++)
    {
        cout << cot[i];
        if (i != res.size()) cout << ' ';
    }
    return 0;
}

 注意事项：

并查集，题目给了3000ms，实际最大用时9ms哈哈哈~

如有问题，欢迎提出。