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

##学习并查集##查阅：http://blog.csdn.net/qq_21899803/article/details/51557517

#include<bits/stdc++.h>
using namespace std;
#define size 1000+10
int rk[size],pre[size];
int cmp(int a,int b){
    return a>b;
}//测试
void Init(){
    for(int i=0;i<size;i++){
        pre[i]= i;
    }
}//测试
void Init1(){
    for(int i=0;i<size;i++){
        printf("%3d ",pre[i]);
        if(i%7==0){
        printf("\n");
        }
    }
    printf("\n");
}
int find(int x){//迭代压缩路径
    //递归
    //return x == pre[x] ? x : pre[x] = find(pre[x]);
    int k, j, r;
    r = x;
    while(r != pre[r])
        r = pre[r];
    k = x;
    while(k != r){
        j = pre[k];
        pre[k] = r;
        k = j;
    }
    return r;
}
void join(int fx,int fy){
    int x=find(fx),y=find(fy);
    if(x!=y){
        pre[x]=y;
    }
}
int main(){

    int n;
    int num,num1,num_;
    //freopen("1.txt","r",stdin);
    cin>>n;
    Init();//Init1();
    for(int i=0;i<n;i++){
        scanf("%d:",&num);
        int t=num;
        scanf("%d",&num1);
        rk[i]=num1;
        while(t-->1){
            scanf("%d",&num_);
            join(num1,num_);
        }
    }
    int root[size]={0},res=0;//根节点,树数
    for(int i = 0; i < n; i++)
        root[find(rk[i])]++;
    //sort(root,root+size,cmp);
    sort(root,root+size,greater<int>());
    for(int i = 0; i < size; i++)
        if(root[i]) res++;
        else break;
    printf("%d\n",res);
    for(int i=0;i<res;i++){
        printf("%d",root[i]);
        if(i!=res-1){
            printf(" ");
        }
    }
    printf("\n");
    return 0;
}