天梯L3-003. 社交集群——并查集

Posted 2020-11-01 i am back

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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 <math.h>
#include <algorithm>
#include <stdio.h>
using namespace std;
int c[1005],f[1005];
int find(int x)
{
    return x==f[x]?x:f[x]=find(f[x]);
}

void merge(int x,int y)
{
    x=find(x),y=find(y);
    if(x!=y) f[x]=y;
}
bool cmp(int x,int y)
{
    return x>y;
}
int main()
{
 //freopen("in.txt","r",stdin);
    int n,k,tt,v;
    cin>>n;
    for(int i=0;i<1005;i++) f[i]=i;
    for(int i=0;i<n;i++)
    {
        scanf("%d: ",&k);
        scanf("%d",&tt);
        c[i]=tt;
        for(int i=1;i<k;i++)
        {
            cin>>v;
            merge(tt,v);
        }
    }
    //for(int i=0;i<n;i++) cout<<c[i]<<" ";
    int num[1005]={0},cnt=0;
    for(int i=0;i<n;i++)
        num[find(c[i])]++;
    sort(num,num+1005,cmp);
    for(int i=0;i<1005;i++)
        if(num[i]!=0) cnt++;
    cout<<cnt<<endl;
    for(int i=0;i<cnt;i++)
    {
        if(i==0) cout<<num[i];
        else cout<<" "<<num[i];
    }
    cout<<endl;
}