给出 n 个节点,标号分别从 0 到 n - 1 并且给出一个 无向 边的列表 (给出每条边的两个顶点), 写一个函数去判断这张`无向`图是否是一棵树
样例
样例 1:
输入: n = 5 edges = [[0, 1], [0, 2], [0, 3], [1, 4]]
输出: true.
样例 2:
输入: n = 5 edges = [[0, 1], [1, 2], [2, 3], [1, 3], [1, 4]]
输出: false.
注意事项
你可以假设我们不会给出重复的边在边的列表当中. 无向边 [0, 1] 和 [1, 0] 是同一条边, 因此他们不会同时出现在我们给你的边的列表当中。
bool validTree(int n, vector<vector<int>> &edges)
{
if (n == 1)
{
return true;
}
if (edges.empty())
{
return false;
}
set<int> intSet;
vector<bool> visited(n + 1, false);
std::queue<int>qKey;
qKey.push(edges[0][0]);
while (!qKey.empty())
{
int node = qKey.front();
intSet.insert(node);
if (visited[node] == true)
{
return false;
}
visited[node] = true;
qKey.pop();
for (int i = 0; i < edges.size(); i++)
{
int first = edges[i][0];
int second = edges[i][1];
if (first == node && visited[second] == false)
{
qKey.push(second);
}
if (second == node && visited[first] == false)
{
qKey.push(first);
}
}
}
if (intSet.size() != n)
{
return false;
}
return true;
}