[图的遍历]--连通图

题目来源：codeup 图的遍历–连通图

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

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

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

样例输入
4 3
4 3
1 2
1 3
5 7
3 5
2 3
1 3
3 2
2 5
3 4
4 1
7 3
6 2
3 1
5 6
0 0
样例输出
YES
YES
NO

#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <cstdio>
#include <vector>
#include <set>
using namespace std;
const int MAXV = 1010; //最大顶点数
int n;                    //顶点数

//DFS：以深度作为第一关键词，每次都是沿着路径走到不能再前进时才退回到最近的岔道口

//邻接表版
vector<int> Adj[MAXV];
bool vis[MAXV] = {false};
void DFS(int u, int depth)
{
    vis[u] = true;
    for (vector<int>::iterator it = Adj[u].begin(); it != Adj[u].end(); it++)
    {
        int v = *it;
        if (!vis[v])
        {
            DFS(v, depth + 1);
        }
    }
    Adj[u].clear();  //这样下一组输入数据就不用初始化了
}

void DFSTrave(int n, int &count)
{
    for (int i = 1; i <= n; i++)
    {
        if (!vis[i])
        {
            count++;
            DFS(i, 1);
        }
    }
}

int main()
{
    int n, m;
    while (cin >> n >> m && n != 0)
    {
        memset(vis, 0, sizeof(vis));
        int u, v;
        for (int i = 0; i < m; i++)
        {
            cin >> u >> v;
            Adj[u].push_back(v);
            Adj[v].push_back(u);
        }
        int count = 0;
        DFSTrave(n, count);
        if (count == 1)
        {
            cout << "YES" << endl;
        }
        else if (count > 1)
        {
            cout << "NO" << endl;
        }
    }

    return 0;
}