每日一题之 hiho1332 树结构的判定

描述
给定一个包含 N 个顶点 M 条边的无向图 G ，判断 G 是不是一棵树。

输入
第一个是一个整数 T ，代表测试数据的组数。 (1 ≤ T ≤ 10)

每组测试数据第一行包含两个整数 N 和 M 。(2 ≤ N ≤ 500, 1 ≤ M ≤ 100000)

以下 M 行每行包含两个整数 a 和 b ，表示顶点 a 和顶点 b 之间有一条边。(1 ≤ a, b ≤ N)

输出
对于每组数据，输出YES或者NO表示 G 是否是一棵树。

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

思路：

树的性质，n个节点只有n-1条边。且是连通的。用并查集来做的话，每次判断两个点是否属于同一个集合。如果不属于则这两个点合并。最后判断每个点，$parent[i] == i$的点有多少个。如果只有一个，说明所有的点位于同一个集合且以这个点为根节点的树，否则不连通就不是树

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>

using namespace std;

const int maxn = 505;

int parent[maxn];
bool visit[maxn];


void init()
{   
    memset(visit,0,sizeof(visit));
    for (int i = 1; i <= maxn; ++i)
        parent[i] = i;
}

int Find(int x)
{
    if (parent[x] != x)
        parent[x] =  Find(parent[x]);
    return parent[x];
}

void Union(int x, int y)
{
    int fx = Find(x);
    int fy = Find(y);

    if (fx != fy)
        parent[fy] = fx;

}


int main()
{
    int t;
    cin >> t;
    while(t--) {
        init();
        int n,m,u,v;
        cin >> n >> m;
        bool flag = true;
        if (m != n-1) flag = false;
        int res = 0;
        for (int i = 0; i < m; ++i) {
            cin >> u >> v;
            Union(u,v);
        }
        for (int i = 1; i <= n; ++i)
            if (parent[i] == i)
                ++res;
            //cout << parent[i] <<endl;

        if (res == 1 && flag)
            cout << "YES" << endl;
        else 
            cout << "NO" << endl;

    }

    return 0;
}