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.
1public class Solution {
2 public int Find(int[] parent, int x) {
3 if (parent[x] != x) {
4 parent[x] = Find(parent, parent[x]);
5 }
6 return parent[x];
7 }
8
9 public bool Union(int[] parent, int x, int y) {
10 int rootX = Find(parent, x);
11 int rootY = Find(parent, y);
12 if (rootX == rootY) return false;
13 parent[rootY] = rootX;
14 return true;
15 }
16
17 public bool ValidateBinaryTreeNodes(int n, int[] leftChild, int[] rightChild) {
18 int[] parent = new int[n];
19 int[] hasParent = new int[n];
20 for (int i = 0; i < n; ++i) parent[i] = i;
21
22 for (int i = 0; i < n; ++i) {
23 if (leftChild[i] != -1) {
24 if (hasParent[leftChild[i]] == 1 || !Union(parent, i, leftChild[i])) {
25 return false;
26 }
27 hasParent[leftChild[i]] = 1;
28 }
29 if (rightChild[i] != -1) {
30 if (hasParent[rightChild[i]] == 1 || !Union(parent, i, rightChild[i])) {
31 return false;
32 }
33 hasParent[rightChild[i]] = 1;
34 }
35 }
36
37 int rootCount = 0;
38 for (int i = 0; i < n; ++i) {
39 if (hasParent[i] == 0) {
40 rootCount++;
41 }
42 }
43
44 return rootCount == 1;
45 }
46}
The C# implementation mirrors other union-find solutions, modifying the state intelligently between nodes. Components track their leaders (or roots) through recursion and path compression in Find
and are unified via Union
. Constraints enforce at most one parent per node via a check through hasParent
, ensuring single connectivity by requiring one and only one root.
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.
1#include <stdbool.h>
2#include <string.h>
3
4void dfs(int node, int* visited, int** children, int n) {
5 if (visited[node]) return;
6 visited[node] = 1;
7 if (children[0][node] != -1) dfs(children[0][node], visited, children, n);
8 if (children[1][node] != -1) dfs(children[1][node], visited, children, n);
9}
10
11bool validateBinaryTreeNodes(int n, int* leftChild, int leftChildSize, int* rightChild, int rightChildSize) {
12 int inDegree[n];
13 memset(inDegree, 0, sizeof(inDegree));
14
15 for (int i = 0; i < n; ++i) {
16 if (leftChild[i] != -1) ++inDegree[leftChild[i]];
17 if (rightChild[i] != -1) ++inDegree[rightChild[i]];
18 }
19
20 int root = -1;
21 for (int i = 0; i < n; ++i) {
22 if (inDegree[i] == 0) {
23 if (root == -1) {
24 root = i;
25 } else {
26 return false; // more than one root
27 }
28 }
29 }
30
31 if (root == -1) return false; // no root
32
33 int visited[n];
34 memset(visited, 0, sizeof(visited));
35 dfs(root, visited, (int*[]){leftChild, rightChild}, n);
36
37 for (int i = 0; i < n; ++i) {
38 if (!visited[i]) return false; // not fully connected
39 }
40
41 return true;
42}
In this C solution, the algorithm starts by calculating the in-degree for each node, which should equal zero for the root node. A DFS checks connectivity from a root node, ensuring all nodes are part of one connected graph and are reachable. Each node is visited once, ensuring an entire traversal defines the single connected component rule of trees.