0에서 n-1까지 레이블이 지정된 n개의 노드가 있고 방향이 지정되지 않은 모서리 [u,v] 목록이 있다고 가정합니다. 이러한 모서리가 유효한 트리를 구성하는지 여부를 확인하는 함수를 정의해야 합니다.
따라서 입력이 n =5이고 edge =[[0,1], [0,2], [0,3], [1,4]]인 경우 출력은 true가 됩니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
-
dfs() 함수를 정의하면 노드, 파, 그래프 및 방문이라는 다른 배열이 필요합니다.
-
방문한[노드]가 1과 같으면 -
-
true를 반환
-
-
방문한[노드]가 2와 같으면 -
-
거짓 반환
-
-
방문[노드] :=2
-
ret :=참
-
초기화 i :=0의 경우, i <그래프[노드]의 크기일 때 업데이트(i를 1만큼 증가), 수행 -
-
그래프[노드, i]가 par와 같지 않으면 -
-
ret :=ret AND dfs(그래프[노드, i], 노드, 그래프, 방문)
-
-
-
방문[노드] :=1
-
리턴 렛
-
주요 방법에서 다음을 수행하십시오 -
-
크기가 n인 방문 배열을 정의하고 0으로 채웁니다.
-
크기가 n인 그래프라는 목록 목록을 정의합니다.
-
initialize i :=0의 경우, i <모서리의 크기일 때 업데이트(i를 1만큼 증가), 수행 -
-
u :=가장자리[i, 0], v :=가장자리[i, 1]
-
그래프 끝에 v 삽입[u]
-
그래프 끝에 u 삽입[v]
-
-
dfs(0, -1, 그래프, 방문)이 거짓이면 -
-
거짓 반환
-
-
initialize i :=0의 경우, i
-
방문[i]이 0이면 -
-
거짓 반환
-
-
-
true를 반환
예시
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
#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