图 连通图（并查集）_59

连通图
【并查集也用来求解是否为连通图的问题，当集合个数为1时，连通分量是1，说明是连通图】
注： 和本专栏的_58题代码几乎一样，只是两个不同的应用场景。
描述
给定一个无向图和其中的所有边，判断这个图是否所有顶点都是连通的。
输入描述：
每组数据的第一行是两个整数 n 和 m（0<=n<=1000）。n 表示图的顶点数目，m 表示图中边的数目。随后有 m 行数据，每行有两个值 x 和 y（0<x, y <=n），表示顶点 x 和 y 相连，顶点的编号从 1 开始计算。输入不保证这些边是否重复。
输出描述：
对于每组输入数据，如果所有顶点都是连通的，输出"YES"，否则输出"NO"。
【示例】
输入：
4 3
1 2
2 3
3 2
3 2
1 2
2 3
0 0

输出：
NO
YES

#include<bits/stdc++.h>
using namespace std;
//并查集 
const int MAXN=1000;
int father[MAXN];
int height[MAXN];
void init(int n){
	for(int i=1;i<=n;i++){
		father[i]=i;
		height[i]=0;
	}
}
//返回父结点 
int find(int x){
	if(x!=father[x]){
		father[x]=find(father[x]);//寻找根结点，压缩路径 
	}
	return father[x];
}
//合并集合 
void union1(int x,int y){
	int x1=find(x);
	int y1=find(y);
	//如果x和y不在一个集合内就合并 
	if(x1!=y1){
		//比较两“树”的高度
		if(height[x1]<height[y1]){
			//x树的高度小于y树的高度
			father[x1]=y1; 
		}
		else if(height[x1]>height[y1]){
			//y树的高度小于x树的高度
			father[y1]=x1; 
		}
		else{
			father[y1]=x1;
			height[x1]++;
			//由于实现了路径压缩，所以height加1就好 
		}
	}
}
int main(){
	int n,m,x,y,number;
	while(cin>>n){
		if(n==0){
			break;
		}
		//初始化每个城镇的父结点是它自己 
		init(n);
		cin>>m;
		while(m--){
			//输入道路 
			cin>>x>>y;
			//合并集合 
			union1(x,y);
		}
		number=0;
		for(int j=1;j<=n;j++){
			if(find(j)==j){
				number++;
			}
		}
		if(number==1){
			cout<<"YES"<<endl;
		}
		else{
			cout<<"NO"<<endl;
		}
	}
	return 0;
}