判断连通图（并查集 ）

给定一个无向图和其中的所有边，判 断这个图是否所有顶点都是连通的。
输入：每组数据的第一行是两个整数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

#include<stdio.h>
int map[1005];
int n,m;
int find(int i)
{
    return map[i]==i?i:find(map[i]);
}
void init()
{
    for(int i=0;i<n;i++)
    map[i]=i;
}
int main()
{
    while(scanf("%d%d",&n,&m)==2)
    {
        if(n==0) break;
        init();
        int a,b;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&a,&b);
            a--;b--;
            map[find(a)]=map[find(b)];
        }
        int cnt=0;
        for(int i=0;i<n;i++)
        {
            if(map[i]==i)
            cnt++;
        }
        if(cnt==1)
        printf("YES\n");
        else
        printf("NO\n");
    }
    return 0;
}