并查集仅有一个根节点,因此设置isRoot[N]记录根节点的数目
#include<cstdio>
const int N=110;
int father[N];
bool isRoot[N];
int findFather(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=findFather(a);
int faB=findFather(b);
if(faA!=faB){
father[faA]=faB;
}
}
void init(int n){
for(int i=1;i<=n;i++){
father[i]=i;
isRoot[i]=false;
}
}
int main(){
int n,m,a,b;
scanf("%d%d",&n,&m);
init(n);//结点个数
for(int i=0;i<m;i++){
scanf("%d%d",&a,&b);
Union(a,b);
}//连接
for(int i=1;i<=n;i++){
isRoot[findFather(i)]=true;
}
int ans=0;//分组个数
for(int i=1;i<=n;i++){
ans+=isRoot[i];
}
printf("%d\n",ans);
return 0;
}
如果要求每个集合中的元素:
#include<cstdio>
const int N=110;
int father[N];
int isRoot[N];
int findFather(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=findFather(a);
int faB=findFather(b);
if(faA!=faB){
father[faA]=faB;
isRoot[faB]+=isRoot[faA]; //当faA==faB时,明明就在一个集合里,是好朋友,不需要改变
isRoot[faA]=0;
}
}
void init(int n){
for(int i=1;i<=n;i++){
father[i]=i;
isRoot[i]=1;
}
}
int main(){
int n,m,a,b;
scanf("%d%d",&n,&m);
init(n);//结点个数
for(int i=0;i<m;i++){
scanf("%d%d",&a,&b);
Union(a,b);
}//连接
for(int i=1;i<=n;i++){
if(isRoot[i]) printf("%d\n",isRoot[i]);
}
return 0;
}