PAT--L3-003社交集群------并查集

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

输入格式：
输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：
K i : h i [1] h i [2] … h i [K i ]
其中K i (>0)是兴趣爱好的个数，h i [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<iostream>
#include<algorithm>
using namespace std;
int father[10001],isRoot[10001]={0};
int find(int x)
{
    int a=x;
    while(x!=father[x]) x=father[x];
    while(a!=father[a]){
        int z=a;
        a=father[a];
        father[z]=x;
    }
    return x;
}
void Union(int a,int b)
{
    int faa=find(a);
    int fab=find(b);
    if(faa<fab) father[fab]=faa;
    else father[faa]=fab;
}
bool cmp(int a,int b){ return a>b; }
int main()
{
    int n,m,k,a[10001]={0},b,flag=0;
    scanf("%d",&n);
    for(int i=1;i<10001;i++) father[i]=i;
    for(int i=1;i<=n;i++){
        scanf("%d:",&m);
        for(int j=0;j<m;j++){
            scanf("%d",&b);
            if(a[b]==0) a[b]=i;
            Union(i,a[b]);
        }
    }
    for(int i=1;i<=n;i++){
        if(isRoot[find(i)]==0) flag++;
        isRoot[find(i)]++;
    }
    sort(isRoot,isRoot+10001,cmp);
    printf("%d\n",flag);
    for(int i=0;i<flag;i++)
        i!=flag-1?printf("%d ",isRoot[i]):printf("%d",isRoot[i]);
    return 0;
}

来总结一下：
1）在其他的并查集题目当中，都是每一个人都有不同的id，而且可以直接读取，所以直接用find,union两个函数直接合并成不同的集合就行。
2）这个题的不同之处，在于，这个题输入的内容是共同的兴趣，而不是每一个有这个兴趣的人的id，所以想要合并的话主要是要先让每一个兴趣都有一个id，然后才可以直接合并，就有了数组a，当其中的值为0时，说明这个兴趣还没有出现过，所以就给它初始化一个，这样就可以直接合并了。

PAT--L3-003社交集群------并查集

猜你喜欢