在C ++中绘制有效树图
假设我们有n个节点,它们被标记为从0到n-1,并列出了无向边[u,v]的列表,我们必须定义一个函数来检查这些边是否构成了有效的树。
因此,如果输入像n=5,并且边沿=[[0,1],[0,2],[0,3],[1,4]],那么输出将为true
为了解决这个问题,我们将遵循以下步骤-
定义一个函数dfs(),它将使用节点,参数,图和另一个称为Visited的数组,
如果visited[node]与1相同,则-
返回真
如果visited[node]与2相同,则-
返回假
visit[node]:=2
ret:=true
对于初始化i:=0,当i<graph[node]的大小时,更新(将i增加1),执行-
ret:=retANDdfs(graph[node,i],node,graph,visited)
如果graph[node,i]不等于par,则-
Visited[node]:=1
返回ret
从主要方法中执行以下操作-
定义一个大小为n的访问数组,并用0填充。
定义一个列表列表,称为大小为n的图
对于初始化i:=0,当i<边的大小时,更新(将i增加1),执行-
u:=edges[i,0],v:=edges[i,1]
在图形的末尾插入v[u]
在图形[v]的末尾插入u
如果dfs(0,-1,graph,Visited)为假,则-
返回假
对于初始化i:=0,当i<n时,更新(将i增加1),执行-
返回假
如果visit[i]为零,则-
返回真
例
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
bool dfs(int node, int par, vector <int< graph[], vector <int<& visited){
if (visited[node] == 1)
return true;
if (visited[node] == 2)
return false;
visited[node] = 2;
bool ret = true;
for (int i = 0; i < graph[node].size(); i++) {
if (graph[node][i] != par)
ret &= dfs(graph[node][i], node, graph, visited);
}
visited[node] = 1;
return ret;
}
bool validTree(int n, vector<vector<int<>& edges) {
vector<int< visited(n, 0);
vector<int< graph[n];
for (int i = 0; i < edges.size(); i++) {
int u = edges[i][0];
int v = edges[i][1];
graph[u].push_back(v);
graph[v].push_back(u);
}
if (!dfs(0, -1, graph, visited))
return false;
for (int i = 0; i < n; i++) {
if (!visited[i])
return false;
}
return true;
}
};
main(){
Solution ob;
vector<vector<int<> v = {{0,1},{0,2},{0,3},{1,4}};
cout << (ob.validTree(5,v));
}输入项
5, {{0,1},{0,2},{0,3},{1,4}}输出结果
1