L3-1 社交集群（30 分）

最新推荐文章于 2022-12-18 01:45:29 发布

jiang_16

最新推荐文章于 2022-12-18 01:45:29 发布

阅读量1.2k

点赞数

分类专栏： PAT 并查集

本文链接：https://blog.csdn.net/jiang_1603/article/details/79715065

版权

PAT 同时被 2 个专栏收录

20 篇文章

订阅专栏

并查集

6 篇文章

订阅专栏

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

输入格式：

输入在第一行给出一个正整数 N（ $\leq 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] 内的整数。

输出格式：

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

输入样例：

输出样例：

3
4 3 1

很明显是并查集，比赛的时候我在想怎么判断两个人有相同的兴趣，，结果还是暴力。。先暴力搜索有相同兴趣的两个人，剩下的就是并查集了

#include <bits/stdc++.h>

using namespace std;
const int N = 1e3 + 3;

int f[N], ch[N];

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

void Union(int x, int y)
{
    int fx = Find(x);
    int fy = Find(y);
    if(fx != fy)
    {
        f[fx] = fy;
        ch[fy] += ch[fx];
    }
}

int main()
{
    int n, k, h;
    scanf("%d", &n);
    vector<set<int> > vec(N);
    for(int i = 1; i <= n; i++)
    {
        f[i] = i;
        ch[i] = 1;
    }
    for(int i = 1; i <= n; i++)
    {
        scanf("%d:", &k);
        while(k--)
        {
            scanf("%d", &h);
            vec[i].insert(h);
            for(int j = 1; j < i; j++)
            {
                if(vec[j].count(h) != 0)
                {
                    Union(i, j);
                    break;
                }
            }
        }
    }
    int cnt = 0, ans[N];
    for(int i = 1; i <= n; i++)
    {
        if(Find(i) == i)
        {
            ans[cnt] = ch[i];
            cnt++;
        }
    }
    sort(ans, ans + cnt);
    printf("%d\n", cnt);
    for(int i = cnt-1; i >= 0; i--)
    {
        printf("%d", ans[i]);
        if(i > 0)
            printf(" ");
    }
    return 0;
}