并查集 代码模板+理解

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;

int isRoot[100];
int father[100]; 

//n个节点的初始化 
void init(int n){
	
	for(int i=1;i<=n;i++){
		father[i]=i;//默认父节点是本身 
		isRoot[i]=0;
	}
}

//寻找x的父节点 并且采用路径压缩的方式 
int findFather(int x){
	int a=x;//先保存起来x节点 后面x改变了 
	while(x!=father[x])
		x=father[x];
	//循环结束的时候 x是原本x的根节点 
	
	cout<<"找到"<<a<<"的根节点为"<<x<<endl;
	cout<<"路径压缩情况:"<<endl;

	while(a!=father[a]){
		int z=a;
		a=father[a];//保存a的父节点
		father[z]=x;//设置z的根节点是x 
		cout<<z<<"的根节点设置为"<<x<<"  ";
	}
	cout<<"..............."<<endl;
	return x;//返回根节点 
} 


//合并a b所在的集合 
void Union(int a,int b){
	
	int fatherA=findFather(a);
	int fatherB=findFather(b);
	
	if(fatherA!=fatherB){
		cout<<a<<" "<<b<<" 不在同一个集合里，将"<<a<<"的跟节点"<<fatherA
		<<"的 新的根节点设置为"<<b<<"的根节点"<<endl;; 
		father[fatherA]=fatherB; 
	}
		
}

int main(void){
	
	int n,m,a,b;
	
	cin>>n>>m;
	init(n);
	for(int i=1;i<=m;i++){
		cin>>a>>b;
		//两者是朋友关系 属于一个集合的 所以合并
		Union(a,b); 
	} 
	
	for(int i=1;i<=n;i++){
		cout<<i<<"的根节点是"<<findFather(i)<<" "; 
		isRoot[findFather(i)]++;//i所在集合的根节点被作为根节点一次 加1说明有一个子节点 
		cout<<"当前isRoot["<<findFather(i)<<"] "<<isRoot[findFather(i)]<<endl;
	}
	
	//sort(isRoot,isRoot+n+1); 
	for(int i=1;i<=n;i++){
		if(isRoot[i]!=0){
			cout<<i<<"  是根节点，该组有"<<isRoot[i]<<" 个节点"<<endl; 
		}
	}
	
	return 0;
}



/*

7 5
1 2
2 3
3 1
1 4
5 6


*/