[hihocoder1322] 树结构判定

这是hiho一下 第161周的题目

---

题目描述

时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
给定一个包含 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

算法简介

首先通过M==N-1做树判断。
然后深度优先搜索，判断是否存在回路，存在回路则非树。
最后遍历所有节点，判断是否有节点没有被访问。

犯错误的部分：因为是无向边，所以添加的时候需要在两个节点都添加；而使用过边之后，需要两个方向的边都删除。

代码

#include <iostream>
#include <list>

using namespace std;

struct Node {
    list<int> next;
};

int N,M,a,b;
bool flag[502];

bool test(Node* nodes,int s) {
    if (flag[s])
        return false;
    flag[s] = true;
    for (list<int>::iterator iter = nodes[s].next.begin();iter != nodes[s].next.end();++iter) {
        nodes[*iter].next.remove(s);
        if (!test(nodes,*iter))
            return false;
    }
    return true;
}

bool deter() {
    cin >> N >> M;
    if (M != N-1) {
        for (int i = 0;i < M;++i)
            cin >> a >> b;
        return false;
    }
    Node nodes[502];
    for (int i = 0;i < M;++i) {
        cin >> a >> b;
        nodes[a-1].next.push_front(b-1);
        nodes[b-1].next.push_front(a-1);
    }
    for (int i = 0;i < N;++i)
        flag[i] = false;
    if (!test(nodes,0))
        return false;
    for (int i = 0;i < N;++i)
        if (!flag[i])
            return false;
    return true;
}

int main()
{
    int T;
    cin >> T;
    for (int i = 0;i < T;++i)
        cout << ((deter())?"YES":"NO") << endl;
    return 0;
}