To solve this problem using the Disjoint Set Union (DSU) approach, we aim to union nodes based on their parent-child relationships. A valid tree should follow these rules:
Time Complexity: O(n), where n is the number of nodes, due to each union and find operation being nearly constant with path compression.
Space Complexity: O(n) for the parent array.
1class Solution:
2 def find(self, parent, x):
3 if parent[x] != x:
4 parent[x] = self.find(parent, parent[x])
5 return parent[x]
6
7 def union(self, parent, x, y):
8 rootX = self.find(parent, x)
9 rootY = self.find(parent, y)
10 if rootX == rootY:
11 return False
12 parent[rootY] = rootX
13 return True
14
15 def validateBinaryTreeNodes(self, n, leftChild, rightChild):
16 parent = list(range(n))
17 hasParent = [0] * n
18
19 for i in range(n):
20 if leftChild[i] != -1:
21 if hasParent[leftChild[i]] or not self.union(parent, i, leftChild[i]):
22 return False
23 hasParent[leftChild[i]] = 1
24 if rightChild[i] != -1:
25 if hasParent[rightChild[i]] or not self.union(parent, i, rightChild[i]):
26 return False
27 hasParent[rightChild[i]] = 1
28
29 return hasParent.count(0) == 1
The Python solution follows the union-find pattern, organized into find
and union
methods. The parent
list captures each node’s representative root, while hasParent
ensures a node has at most one parent. Disjoint set operations verify no cycles and ensure all nodes unify under one connected root without duplication errors, confirming only one root node.
This approach involves calculating the in-degree of each node and checking connectivity via a DFS. The key aspects of a tree like single-root presence and cycle-checking can be managed by:
Time Complexity: O(n), since each node and its immediate edges are evaluated once in each step, including in-drives calculations and DFS.
Space Complexity: O(n) for holding visited tracking and in-degree counts.
1class Solution:
2 def dfs(self, node, visited, children):
3 if visited[node]: return
4 visited[node] = True
5 if children[0][node] != -1:
6 self.dfs(children[0][node], visited, children)
7 if children[1][node] != -1:
8 self.dfs(children[1][node], visited, children)
9
10 def validateBinaryTreeNodes(self, n, leftChild, rightChild):
11 # Calculate in-degrees
12 inDegree = [0] * n
13 for i in range(n):
14 if leftChild[i] != -1:
15 inDegree[leftChild[i]] += 1
16 if rightChild[i] != -1:
17 inDegree[rightChild[i]] += 1
18
19 root = -1
20 for i in range(n):
21 if inDegree[i] == 0:
22 if root == -1:
23 root = i
24 else:
25 return False
26
27 if root == -1:
28 return False
29
30 visited = [False] * n
31 self.dfs(root, visited, [leftChild, rightChild])
32
33 return all(visited)
The Python strategem frames nodes as tuples within a list structure, exploiting recognized semantics over positions within binary arrays (left and right children). Node in-degrees determine the hierarchy without multi-parents obstruction, generating a singular possible valid root. DFS confirms full graph traversability from the root node, ensuring a connected nature without interruptions.