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<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
int a[1001][1001],f[1005],w[1005],c[1005];
int fine(int v)
{
    if(f[v]==v) return v;
    else
    {
        f[v]=fine(f[v]);
        return f[v];
    }
}
void bcj(int v,int u)
{
    int t1,t2;
    t1=fine(v);
    t2=fine(u);
    if(t1!=t2)
        f[t2]=t1;
}
bool cmp(int v,int m)
{
    return v>m;
}

int main()
{
    int n,i,j,x=0;
    scanf("%d",&n);
    for(i=0; i<=1000; i++)
    {
        f[i]=i;
        w[i]=1;
    }
    for(i=0; i<n; i++)
    {
        scanf("%d: ",&a[i][0]);
        for(j=1; j<=a[i][0]; j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    for(i=0; i<n; i++)
    {
        for(j=2; j<=a[i][0]; j++)
        {
            bcj(a[i][j-1],a[i][j]);
        }
    }
    for(i=0; i<n; i++)
    {
        if(w[i]==0) continue;
        c[x]=1;
        w[i]=0;
        for(j=i+1; j<n; j++)
        {
            if(w[j]==0) continue;
            if(fine(a[i][1])==fine(a[j][1]))
            {
                c[x]++;
                w[j]=0;
            }
        }
        x++;
    }
    printf("%d\n",x);
    sort(c,c+x,cmp);
    for(i=0; i<x; i++)
    {
        printf("%d",c[i]);
        if(i!=x-1) printf(" ");
    }
    printf("\n");
    return 0;
}