PAT甲题题解-1107. Social Clusters (30)-PAT甲级真题（并查集）

　　题意：有n个人，每个人有k个爱好，如果两个人有某个爱好相同，他们就处于同一个集合。问总共有多少个集合，以及每个集合有多少人，并按从大到小输出。

　　很明显，采用并查集。vis[k]标记爱好k第一次出现的人的编号，如果为0则表示未出现。

　　当前第i个人若也存在爱好k，则只要将i与vis[k]两个人合并即可。

最后father[i]相同的即处在同一个集合中。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
#include <cmath>
using namespace std;
/*
并查集
*/
const int maxn=1005;
int vis[maxn]; //vis[i]标记某爱好第一次出现在第几个人，0表示还未出现。
int n;

struct UF{
int father[maxn];
void init(){
for(int i=0;i<maxn;i++){
father[i]=i;
}
}
int find_root(int x){
if(father[x]!=x){
father[x]=find_root(father[x]);
}
return father[x];
}
void Union(int x,int y){
int fx=find_root(x);
int fy=find_root(y);
if(fx!=fy){
father[fy]=fx;
}
}
}uf;

bool cmp(int a,int b){
return a>b;
}
int main()
{
char str[20];
int num,a;
scanf("%d",&n);
memset(vis,-1,sizeof(vis));
uf.init();
for(int i=1;i<=n;i++){
scanf("%s",str);
int len=strlen(str);
str[len-1]='\0';
num=atoi(str);
for(int k=0;k<num;k++){
scanf("%d",&a);
if(vis[a]==-1)
vis[a]=i;
else{
uf.Union(vis[a],i); //若已出现过，则第i个人与第vis[a]个人合并，即分为一组
}
}
}
int cnt[n+1];
int res=0;
memset(cnt,0,sizeof(cnt));
for(int i=1;i<=n;i++){
int idx=uf.find_root(i);
if(cnt[idx]==0)
res++;
cnt[idx]++;
}
sort(cnt+1,cnt+n+1,cmp);
printf("%d\n",res);
for(int i=1;i<res;i++)
printf("%d ",cnt[i]);
printf("%d",cnt[res]);
return 0;
}

View Code