连通图

题目描述
给定一个无向图和其中的所有边，判断这个图是否所有顶点都是连通的。
输入描述:

每组数据的第一行是两个整数 n 和 m（0<=n<=1000）。n 表示图的顶点数目，m 表示图中边的数目。随后有 m 行数据，每行有两个值 x 和 y（0<x, y <=n），表示顶点 x 和 y 相连，顶点的编号从 1 开始计算。输入不保证这些边是否重复。

输出描述:

对于每组输入数据，如果所有顶点都是连通的，输出"YES"，否则输出"NO"。

示例1
输入
复制

4 3
1 2
2 3
3 2
3 2
1 2
2 3
0 0

输出
复制

NO
YES

#include<iostream>
#define maxsize 1001
using namespace std;
void init();
int find(int i);
void merge(int i,int j);
int n,m,father[maxsize],height[maxsize],v1,v2;
int main(){
    while(cin>>n>>m){
        if(n==0)break;
        if(n==1){
            cout<<"YES"<<endl;
            continue;
        }
        init();
        for(int i=0;i<m;i++){
            cin>>v1>>v2;
            merge(v1,v2);
        }
        int root=find(1),res=1;
        for(int i=2;i<n+1;i++){
            if(find(i)!=root){
                cout<<"NO"<<endl;
                res=0;
                break;
            }
        }
        if(res)cout<<"YES"<<endl;
    }
    return 0;
}
void init(){
    for(int i=1;i<n+1;i++){
        father[i]=i;
        height[i]=0;
    }
}
int find(int i){
    if(father[i]==i){
        return i;
    }
    else{
        father[i]=find(father[i]);
        return father[i];
    }
}
void merge(int i,int j){
    int x=find(i),y=find(j);
    if(x!=y){
        if(height[x]>height[y])father[y]=x;
        else if(height[x]<height[y])father[y]=x;
        else{
            father[x]=y;
            height[y]++;
        }
    }
}