【慢慢学算法】：连通图

题目描述：

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

输入：

    每组数据的第一行是两个整数 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<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

int a[1001][1001];
int visit[1001];
int n, m;

void DFS(int i)
{
    int j;
    visit[i] = 1;
    for(int j = 2; j <= n; j++)
    {
	if( a[i][j] == 1 && visit[j] == 0)
	    DFS(j);
    }
}

int main()
{
    freopen("a.in","r", stdin);
    while( cin >> n >> m)
    {
	int A, B;
	bool flag;
	if( n == 0 && m == 0)
	    break;
	memset(visit, 0, sizeof(visit));
	memset(a, 0, sizeof(a));
	for( int i = 0; i < m; i++)
	{
	    cin >> A >> B;
	    a[A][B] = 1;
	    a[B][A] = 1;
	}
	DFS(1);
	flag = true;
	for(int i = 1; i <= n; i++)
	    if(visit[i] == 0)
		flag = false;
	if(flag)
	    cout << "YES" << endl;
	else
	    cout << "NO" << endl;

    }
    return 0;
}