L3-003. 社交集群

L3-003. 社交集群

时间限制

1000 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入的第一行给出正整数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 <bits/stdc++.h>

using namespace std;
const int maxn=1005;
int father[maxn];
void init()
{
    for(int i=0;i<maxn;i++)
    {
        father[i]=i;
    }
}
int findfather(int x)
{
    if(x==father[x]) return x;
    else  return(father[x]=findfather(father[x]));
}
void hb(int a,int b)
{
    int a1=findfather(a);
    int b1=findfather(b);
    if(a1!=b1) father[a1]=b1;
}
int num[maxn];
int hashtable[maxn];

int main()
{
    init();
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        int sn;
        scanf("%d: ",&sn);

            int  tmp,head;
            scanf("%d",&head);
            num[i]=head;
            for(int k=1;k<sn;k++)
            {
                scanf("%d",&tmp);
                hb(tmp,head);
            }
    }

    for(int i=0;i<n;i++)
    {
        hashtable[findfather(num[i])]++;
    }
    sort(hashtable,hashtable+maxn,greater<int>());
    int cnt=0;
    for(int i=0;i<maxn;i++)
    {

        if(hashtable[i]==0) break;
        cnt++;
    }
    cout<<cnt<<endl;
    for(int i=0;i<cnt;i++)
    {
        cout<<hashtable[i];
        if(i!=cnt-1) cout<<' ';
    }

    return 0;
}