【PTA 天梯赛】L3-003 社交集群（并查集）

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

输入格式：

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

Ki: hi[1] hi[2] ... hi[Ki]

其中Ki(>0)是兴趣爱好的个数，hi[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

#include<bits/stdc++.h>

using namespace std;

int p[],ai[],cir[];

int find(int r)

{

    if(p[r]!=r)

        p[r]=find(p[r]);

    return p[r];

}

void join(int x,int y)

{

    int fx=find(x),fy=find(y);

    if(fx!=fy)

        p[fx]=fy;

}

int main()

{

    int n,i,j,k,x;

    char ch;

    cin>>n;

    for(i=;i<=;i++)

        p[i]=i;

    for(i=;i<=n;i++)

    {

        cin>>k>>ch;

        for(j=;j<=k;j++)

        {

            cin>>x;

            if(ai[x]==)ai[x]=i;//i号人作为爱好x的父亲

            join(i,find(ai[x]));//将i号人与爱好x的人群连接

        }

    }

    int sum=;

    for(i=;i<=n;i++)

        cir[find(i)]++;

    sort(cir+,cir+,greater<int>());//按从大到小排序

    for(i=;i<=;i++)

    {

        if(cir[i]==)

        {

            i--;

            break;

        }

    }

    cout<<i<<endl;

    for(j=;j<=i;j++)

    {

        if(j==)cout<<cir[j];

        else cout<<" "<<cir[j];

    }

    cout<<endl;

    return ;

}

巴特西

【PTA 天梯赛】L3-003 社交集群（并查集）

最新文章

热门文章