九度OJ 1109：连通图

题目描述：

    给定一个无向图和其中的所有边，判断这个图是否所有顶点都是连通的。

输入：

    每组数据的第一行是两个整数 n 和 m（0<=n<=1000）。n 表示图的顶点数目，m 表示图中边的数目。如果 n 为 0 表示输入结束。随后有 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

来源：

2011年吉林大学计算机研究生机试真题

#include <cstdio>
#define N 1002
using namespace std;
int tree[N];
int findRoot(int x){
    int ret = x;
    while(tree[ret] != -1)
        ret = tree[ret];
    while(tree[x] != -1){
        int t = tree[x];
        tree[x] = ret;
        x = t;
    }
    return ret;
}

int main(){
    int n,m;
    while(scanf("%d",&n) != EOF){
        if(n == 0)break;
        scanf("%d",&m);
        for(int i = 1;i <= n;i++)
            tree[i] = -1;
        for(int i = 1;i <= m;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            int a = findRoot(x);
            int b = findRoot(y);
            if(a != b){
                tree[b] = a;
            }
        }
        int cnt = 0;
        for(int i = 1;i <= n;i++)
           if(tree[i] == -1)cnt++;

        printf("%s\n",cnt == 1 ? "YES" : "NO");
    }

    return 0;
}