Data Structures & AlgorithmsGraphs

Trees as Special Graphs

LevelIntermediate

Duration60 mins

TopicGraphs

4 / 4

Forest: Multiple Disconnected Trees

When Trees Grow Together

A tree, by definition, is a connected acyclic graph. But what happens when we relax the connectivity requirement? We get a forest —a collection of trees growing side by side, each isolated from the others. Forests aren't merely academic constructs. They appear naturally in many scenarios: disconnected network components, intermediate states of algorithms building spanning trees, equivalence class partitions in Union-Find, and hierarchical clusterings where not everything is related. Understanding forests completes our picture of acyclic graph structures.

What You Will Learn

By the end of this page, you will understand the formal definition of forests, learn key properties distinguishing forests from trees and general graphs, see how forests arise naturally in algorithms like Kruskal's MST, and master techniques for working with forests—counting components, merging trees, and decomposing graphs.

Formal Definition of a Forest

In graph theory, a forest is defined as follows: Definition: A forest is an undirected graph that is acyclic. Equivalently, a forest is a graph where every connected component is a tree. The key distinctions from a tree: Tree: Connected AND acyclic Forest: Acyclic (but not necessarily connected) Every tree is a forest, but not every forest is a tree. A tree is a forest with exactly one connected component.

Converting Mermaid diagram...

In the diagram above: The forest has 7 vertices It contains 3 trees (connected components) Tree 1 has 4 vertices and 3 edges Tree 2 has 2 vertices and 1 edge Tree 3 has 1 vertex and 0 edges (a single isolated vertex is a valid tree!) Total edges: 4 Key Formula for Forests: If a forest has n vertices and k connected components (trees), then it has exactly n - k edges . Proof: Each of the k trees with nᵢ vertices has nᵢ - 1 edges. Total edges = \sum(nᵢ - 1) = \sumnᵢ - k = n - k.

Forest as Generalization

A tree is a forest with k = 1 component, having n - 1 edges. A forest with n vertices and n - k edges has k components. This formula is incredibly useful for counting components or verifying forest structure.

Properties of Forests

Forests inherit and extend many properties from trees. Understanding these helps in algorithm design and analysis.

Forest Properties
Property	Description	Formula/Value
Components	Number of trees in the forest	k = n - \|E\|
Edges	Total edges in the forest	\|E\| = n - k
Acyclic	No cycles exist	Always true
Intra-component paths	Unique path within each tree	Exactly one (if same tree)
Inter-component paths	Path between trees	None (disconnected)
Max edges	Maximum edges while staying forest	n - 1 (becomes a tree)
Min edges	Minimum edges (empty forest)	0 (all isolated vertices)

Key Structural Properties

•Adding an edge between trees merges them: If u and v are in different trees, adding edge (u, v) reduces component count by 1
•Adding an edge within a tree creates a cycle: If u and v are in the same tree, adding edge (u, v) violates the acyclic property
•Removing any edge increases components: Deleting edge (u, v) splits one tree into two
•Forests are bipartite: Every forest can be 2-colored (vertices colored so no edge connects same colors)
•Forests are planar: Can be drawn without edge crossings

forest_properties.py
Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
from collections import defaultdict
from typing import List, Tuple, Set
 
def analyze_forest(n: int, edges: List[Tuple[int, int]]) -> dict:
    """
    Analyze properties of a forest.
    
    Returns component count, sizes, and validates forest structure.
    """
    # Build adjacency list
    adj = defaultdict(list)
    for u, v in edges:
        adj[u].append(v)
        adj[v].append(u)
    
    # Find connected components via DFS
    visited = set()
    components = []
    
    def dfs(node: int) -> Set[int]:
        """Return all vertices in this component"""
        component = {node}
        stack = [node]
        while stack:
            curr = stack.pop()
            for neighbor in adj[curr]:
                if neighbor not in component:
                    component.add(neighbor)
                    stack.append(neighbor)
        return component
    
    for v in range(n):
        if v not in visited:
            component = dfs(v)
            components.append(component)
            visited.update(component)
    
    # Verify forest property (no cycles): |E| = n - k
    k = len(components)
    expected_edges = n - k
    actual_edges = len(edges)
    is_valid_forest = (actual_edges == expected_edges)
    
    return {
        "vertices": n,
        "edges": actual_edges,
        "components": k,
        "component_sizes": [len(c) for c in components],
        "is_valid_forest": is_valid_forest,
        "edge_formula_check": f"|E|={actual_edges}, n-k={expected_edges}"
    }
 
# Example: A valid forest
forest_edges = [(0, 1), (1, 2), (3, 4)]  # 5 vertices, 2 components
print("Forest analysis:")
print(analyze_forest(5, forest_edges))
 
# Example: Adding an edge that would create a cycle
def would_create_cycle(n: int, edges: List[Tuple[int, int]], new_edge: Tuple[int, int]) -> bool:
    """Check if adding new_edge would create a cycle (violate forest property)"""
    adj = defaultdict(list)
    for u, v in edges:
        adj[u].append(v)
        adj[v].append(u)
    
    # If both endpoints are in the same component, adding the edge creates a cycle
    u, v = new_edge
    
    # DFS from u to see if we can reach v
    visited = {u}
    stack = [u]
    while stack:
        curr = stack.pop()
        if curr == v:
            return True  # u and v already connected → cycle
        for neighbor in adj[curr]:
            if neighbor not in visited:
                visited.add(neighbor)
                stack.append(neighbor)
    
    return False  # u and v in different components → no cycle
 
print("
Would (0, 2) create cycle?", would_create_cycle(5, forest_edges, (0, 2)))  # True
print("Would (2, 3) create cycle?", would_create_cycle(5, forest_edges, (2, 3)))  # False

Forests in Algorithms: Natural Occurrences

Forests appear naturally in many fundamental algorithms. Understanding these occurrences deepens your algorithmic intuition.

Algorithms Where Forests Appear

•Kruskal's MST Algorithm: Starts with n isolated vertices (a forest of n trivial trees), adds edges greedily while maintaining the forest property
•Union-Find (Disjoint Set Union): Represents disjoint sets as trees; the entire structure is a forest where each tree represents one set
•DFS/BFS Forest: When traversing a disconnected graph, the collection of traversal trees forms a forest
•Spanning Forest: For disconnected graphs, a spanning forest is the analog of a spanning tree—covers all vertices with minimum edges per component
•Graph Decomposition: Removing edges from a graph often produces intermediate forest states

Kruskal's Algorithm and Forests

Kruskal's algorithm is the quintessential 'forest-aware' algorithm. It explicitly maintains a forest invariant: at every step, the selected edges form a forest. Each edge addition either merges two trees (safe) or would create a cycle (rejected). The algorithm terminates when exactly one tree remains—the MST.

kruskal_forest.py
Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
from typing import List, Tuple
 
class UnionFind:
    """
    Union-Find data structure for Kruskal's algorithm.
    
    The internal representation is a forest! Each set is a tree.
    """
    def __init__(self, n: int):
        self.parent = list(range(n))
        self.rank = [0] * n
        self.components = n  # Number of trees in the forest
    
    def find(self, x: int) -> int:
        """Find root of tree containing x (with path compression)"""
        if self.parent[x] != x:
            self.parent[x] = self.find(self.parent[x])
        return self.parent[x]
    
    def union(self, x: int, y: int) -> bool:
        """
        Merge trees containing x and y.
        Returns True if merged (were different trees).
        Returns False if same tree (would create cycle).
        """
        root_x, root_y = self.find(x), self.find(y)
        
        if root_x == root_y:
            return False  # Same tree → adding edge would create cycle
        
        # Union by rank
        if self.rank[root_x] < self.rank[root_y]:
            root_x, root_y = root_y, root_x
        self.parent[root_y] = root_x
        if self.rank[root_x] == self.rank[root_y]:
            self.rank[root_x] += 1
        
        self.components -= 1  # Merged two trees into one
        return True
 
def kruskal_mst(n: int, edges: List[Tuple[int, int, int]]) -> Tuple[List, int]:
    """
    Kruskal's algorithm: Build MST by maintaining a forest.
    
    edges: List of (u, v, weight)
    Returns: (MST edges, total weight)
    
    The algorithm explicitly uses the forest property:
    - Start: Forest of n trivial trees (n components)
    - Each step: Add cheapest edge that doesn't create a cycle
    - End: Forest becomes a single tree (the MST)
    """
    # Sort edges by weight
    sorted_edges = sorted(edges, key=lambda e: e[2])
    
    uf = UnionFind(n)
    mst_edges = []
    total_weight = 0
    
    print(f"Starting with forest of {n} isolated vertices (trivial trees)")
    
    for u, v, weight in sorted_edges:
        if uf.union(u, v):
            mst_edges.append((u, v, weight))
            total_weight += weight
            print(f"  Added edge ({u}, {v}) weight {weight} — now {uf.components} tree(s)")
            
            if uf.components == 1:
                print("  Forest became a single tree (MST complete)")
                break
        else:
            print(f"  Skipped edge ({u}, {v}) — would create cycle (same tree)")
    
    return mst_edges, total_weight
 
# Example
edges = [
    (0, 1, 4), (0, 2, 3), (1, 2, 1), (1, 3, 2),
    (2, 3, 4), (3, 4, 2), (2, 4, 5)
]
mst, weight = kruskal_mst(5, edges)
print(f"
MST edges: {mst}")
print(f"Total MST weight: {weight}")

Spanning Forests: Covering Disconnected Graphs

Just as a spanning tree covers all vertices of a connected graph with minimum edges, a spanning forest covers all vertices of a possibly disconnected graph. Definition: A spanning forest of a graph G = (V, E) is a subgraph that: Includes all vertices of G Is a forest (acyclic) Has the same connected components as G (within each component, forms a spanning tree) For a graph with n vertices and k connected components, a spanning forest has exactly n - k edges —the minimum possible while maintaining connectivity within each component.

Spanning Tree (Connected Graph)

•Requires connected graph as input
•Contains exactly 1 tree
•n - 1 edges
•All vertices reachable from any vertex
•BFS/DFS from any start covers all

Spanning Forest (Any Graph)

•Works for any graph (connected or not)
•Contains k trees (k = components)
•n - k edges
•Vertices reachable only within same component
•Multiple BFS/DFS starts may be needed

spanning_forest.py
Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from collections import defaultdict, deque
from typing import List, Tuple, Set
 
def build_spanning_forest(n: int, edges: List[Tuple[int, int]]) -> List[Tuple[int, int]]:
    """
    Build a spanning forest for a (possibly disconnected) graph.
    
    Uses BFS from each unvisited vertex to construct spanning trees
    for each connected component.
    
    Returns: List of edges forming the spanning forest
    """
    adj = defaultdict(list)
    for u, v in edges:
        adj[u].append(v)
        adj[v].append(u)
    
    visited = set()
    spanning_forest_edges = []
    num_trees = 0
    
    def bfs_tree(start: int) -> List[Tuple[int, int]]:
        """Build spanning tree for component containing 'start'"""
        tree_edges = []
        queue = deque([start])
        visited.add(start)
        
        while queue:
            node = queue.popleft()
            for neighbor in adj[node]:
                if neighbor not in visited:
                    visited.add(neighbor)
                    tree_edges.append((node, neighbor))
                    queue.append(neighbor)
        
        return tree_edges
    
    for v in range(n):
        if v not in visited:
            tree_edges = bfs_tree(v)
            spanning_forest_edges.extend(tree_edges)
            num_trees += 1
            print(f"Tree {num_trees}: root at {v}, edges: {tree_edges}")
    
    print(f"
Spanning forest: {num_trees} trees, {len(spanning_forest_edges)} edges")
    print(f"Formula check: n - k = {n} - {num_trees} = {n - num_trees}")
    
    return spanning_forest_edges
 
# Example: A disconnected graph
n = 7
edges = [
    (0, 1), (0, 2), (1, 2),  # Triangle (component 1)
    (3, 4), (4, 5),          # Path (component 2)
    # Vertex 6 is isolated (component 3)
]
 
forest = build_spanning_forest(n, edges)
print(f"
Spanning forest edges: {forest}")

Minimum Spanning Forest

For weighted disconnected graphs, the 'minimum spanning forest' is the spanning forest with minimum total edge weight. Kruskal's algorithm naturally produces this—it doesn't require the input graph to be connected. When the algorithm terminates with k > 1 components, you have a minimum spanning forest instead of a tree.

Common Operations on Forests

Working with forests involves operations on their component trees and managing the overall structure. Here are the essential operations:

Forest Operations
Operation	Description	Complexity
Count components	Determine number of trees	O(V + E) via DFS/BFS
Find component	Which tree contains vertex v?	O(V + E) or O(α(n)) with Union-Find
Same component?	Are u and v in the same tree?	O(α(n)) with Union-Find
Merge trees	Connect two trees with an edge	O(α(n)) with Union-Find
Split tree	Remove an edge, creating two trees	O(V) or O(log V) with advanced structures
Component sizes	Size of each tree	O(V + E)
Total edges check	Verify forest property: \|E\| = n - k	O(1) given counts

forest_operations.py
Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from typing import List, Tuple, Optional
 
class Forest:
    """
    A dynamic forest data structure supporting merge and queries.
    
    Uses Union-Find for efficient component management.
    Tracks component sizes and validates forest invariants.
    """
    def __init__(self, n: int):
        self.n = n
        self.parent = list(range(n))
        self.rank = [0] * n
        self.size = [1] * n  # Size of each component
        self.num_components = n
        self.num_edges = 0
    
    def find(self, x: int) -> int:
        """Find root of tree containing x"""
        if self.parent[x] != x:
            self.parent[x] = self.find(self.parent[x])
        return self.parent[x]
    
    def same_tree(self, u: int, v: int) -> bool:
        """Check if u and v are in the same tree"""
        return self.find(u) == self.find(v)
    
    def tree_size(self, v: int) -> int:
        """Get size of tree containing v"""
        return self.size[self.find(v)]
    
    def add_edge(self, u: int, v: int) -> bool:
        """
        Add edge between u and v.
        Returns True if successful (trees merged).
        Returns False if would create cycle (same tree).
        """
        root_u, root_v = self.find(u), self.find(v)
        
        if root_u == root_v:
            return False  # Would create cycle
        
        # Union by rank
        if self.rank[root_u] < self.rank[root_v]:
            root_u, root_v = root_v, root_u
        
        self.parent[root_v] = root_u
        self.size[root_u] += self.size[root_v]
        
        if self.rank[root_u] == self.rank[root_v]:
            self.rank[root_u] += 1
        
        self.num_components -= 1
        self.num_edges += 1
        return True
    
    def get_components(self) -> List[List[int]]:
        """Get all trees (connected components) in the forest"""
        components = {}
        for v in range(self.n):
            root = self.find(v)
            if root not in components:
                components[root] = []
            components[root].append(v)
        return list(components.values())
    
    def is_single_tree(self) -> bool:
        """Check if forest has become a single tree"""
        return self.num_components == 1
    
    def validate(self) -> bool:
        """Verify forest invariant: |E| = n - k"""
        return self.num_edges == self.n - self.num_components
    
    def __str__(self) -> str:
        return (f"Forest(n={self.n}, edges={self.num_edges}, "
                f"components={self.num_components}, valid={self.validate()})")
 
# Example usage
forest = Forest(6)
print("Initial:", forest)
 
operations = [(0, 1), (2, 3), (4, 5), (0, 2), (3, 4)]
for u, v in operations:
    success = forest.add_edge(u, v)
    status = "merged" if success else "REJECTED (cycle)"
    print(f"Add ({u}, {v}): {status}")
    print(f"  {forest}")
    print(f"  Components: {forest.get_components()}")
 
print(f"
Is single tree? {forest.is_single_tree()}")

Rooted Forests

Just as we can root a tree, we can root a forest by designating a root in each component tree. Definition: A rooted forest is a forest where each tree has a designated root, inducing parent-child relationships on all edges within that tree. Rooted forests are particularly common in: Union-Find implementations: Each set is represented as a rooted tree (with path compression and union-by-rank) Multi-rooted tree problems: When problems have multiple 'starting points' Hierarchical clustering: Multiple disconnected hierarchies

rooted_forest.py
Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from collections import defaultdict
from typing import List, Tuple, Dict
 
def create_rooted_forest(
    n: int, 
    edges: List[Tuple[int, int]], 
    roots: List[int]
) -> Dict[int, List[int]]:
    """
    Root a forest at specified root vertices.
    
    Each connected component should have exactly one root.
    Returns children list for the rooted forest.
    """
    adj = defaultdict(list)
    for u, v in edges:
        adj[u].append(v)
        adj[v].append(u)
    
    children = defaultdict(list)
    visited = set()
    
    def dfs_root(node: int, parent: int):
        """DFS to assign children based on rooting"""
        visited.add(node)
        for neighbor in adj[node]:
            if neighbor not in visited:
                children[node].append(neighbor)
                dfs_root(neighbor, node)
    
    # Root each component from its designated root
    for root in roots:
        if root not in visited:
            dfs_root(root, -1)
    
    # Handle any vertices not reached (isolated or unspecified roots)
    for v in range(n):
        if v not in visited:
            dfs_root(v, -1)  # This vertex becomes root of its component
    
    return dict(children)
 
def print_rooted_forest(children: Dict[int, List[int]], n: int):
    """Pretty print a rooted forest"""
    # Find roots (vertices with no parent)
    has_parent = set()
    for node, kids in children.items():
        has_parent.update(kids)
    
    roots = [v for v in range(n) if v not in has_parent]
    
    def print_tree(node: int, indent: int = 0):
        print("  " * indent + f"└── {node}")
        for child in children.get(node, []):
            print_tree(child, indent + 1)
    
    for root in roots:
        print(f"Tree rooted at {root}:")
        print_tree(root)
        print()
 
# Example: Forest with 2 trees
n = 7
edges = [(0, 1), (1, 2), (1, 3), (4, 5), (5, 6)]
# Tree 1: 0-1-2, 1-3 (rooted at 0)
# Tree 2: 4-5-6 (rooted at 4)
# Vertex 6 would be isolated if it existed without edges
 
rooted = create_rooted_forest(n, edges, roots=[0, 4])
print("Rooted forest structure:")
print_rooted_forest(rooted, n)

Union-Find as Rooted Forest

Union-Find (Disjoint Set Union) maintains each set as a rooted tree, with the root as the 'representative' of the set. Path compression and union-by-rank optimizations ensure the trees stay shallow, giving nearly O(1) amortized operations. The entire data structure is a rooted forest that changes dynamically as sets are merged.

Real-World Applications of Forests

Forests appear in numerous practical contexts, often representing systems where not everything is directly connected.

Forest Applications

•Network Partitions: When a network splits into disconnected components (server clusters, geographic regions), the structure is a forest
•Collaborative Filtering: User-item graphs often have multiple disconnected components representing separate user communities
•Phylogenetics: Evolutionary trees for multiple unrelated lineages form a phylogenetic forest
•Random Forests (ML): Ensemble of decision trees; each tree is independent, together forming a forest of predictors
•Social Network Clusters: Friend networks often have isolated communities—a forest of social trees
•Version Control: Multiple unrelated branches or entirely separate histories form a forest of commit trees
•Component Analysis: Finding connected components in reliability analysis, circuit analysis, or image segmentation

Case Study: Network Reliability In network reliability analysis, we might start with a connected network (a spanning tree of the original graph). As edges fail, the network becomes a forest. Key questions include: How many components remain after k edge failures? Which edges, if failed, would create the most disconnection? What is the probability that two nodes remain connected? These questions reduce to forest analysis: counting components, tracking sizes, and understanding the residual structure.

Forest vs. Tree Mindset

When solving graph problems, don't assume connectivity! Many real-world graphs are disconnected, requiring forest-aware algorithms. Always check: Does my algorithm handle multiple components? Do I need a spanning tree or a spanning forest?

4 / 4

Forest Properties

Property

Description

Formula/Value

Components

Number of trees in the forest

k = n - |E|

Edges

Total edges in the forest

|E| = n - k

Acyclic

No cycles exist

Always true

Intra-component paths

Unique path within each tree

Exactly one (if same tree)

Inter-component paths

Path between trees

None (disconnected)

Max edges

Maximum edges while staying forest

n - 1 (becomes a tree)

Min edges

Minimum edges (empty forest)

0 (all isolated vertices)

from collections import defaultdict from typing import List, Tuple, Set def analyze_forest(n: int, edges: List[Tuple[int, int]]) -> dict: """ Analyze properties of a forest. Returns component count, sizes, and validates forest structure. """ # Build adjacency list adj = defaultdict(list) for u, v in edges: adj[u].append(v) adj[v].append(u) # Find connected components via DFS visited = set() components = [] def dfs(node: int) -> Set[int]: """Return all vertices in this component""" component = {node} stack = [node] while stack: curr = stack.pop() for neighbor in adj[curr]: if neighbor not in component: component.add(neighbor) stack.append(neighbor) return component for v in range(n): if v not in visited: component = dfs(v) components.append(component) visited.update(component) # Verify forest property (no cycles): |E| = n - k k = len(components) expected_edges = n - k actual_edges = len(edges) is_valid_forest = (actual_edges == expected_edges) return { "vertices": n, "edges": actual_edges, "components": k, "component_sizes": [len(c) for c in components], "is_valid_forest": is_valid_forest, "edge_formula_check": f"|E|={actual_edges}, n-k={expected_edges}" } # Example: A valid forest forest_edges = [(0, 1), (1, 2), (3, 4)] # 5 vertices, 2 components print("Forest analysis:") print(analyze_forest(5, forest_edges)) # Example: Adding an edge that would create a cycle def would_create_cycle(n: int, edges: List[Tuple[int, int]], new_edge: Tuple[int, int]) -> bool: """Check if adding new_edge would create a cycle (violate forest property)""" adj = defaultdict(list) for u, v in edges: adj[u].append(v) adj[v].append(u) # If both endpoints are in the same component, adding the edge creates a cycle u, v = new_edge # DFS from u to see if we can reach v visited = {u} stack = [u] while stack: curr = stack.pop() if curr == v: return True # u and v already connected → cycle for neighbor in adj[curr]: if neighbor not in visited: visited.add(neighbor) stack.append(neighbor) return False # u and v in different components → no cycle print(" Would (0, 2) create cycle?", would_create_cycle(5, forest_edges, (0, 2))) # True print("Would (2, 3) create cycle?", would_create_cycle(5, forest_edges, (2, 3))) # False

from typing import List, Tuple class UnionFind: """ Union-Find data structure for Kruskal's algorithm. The internal representation is a forest! Each set is a tree. """ def __init__(self, n: int): self.parent = list(range(n)) self.rank = [0] * n self.components = n # Number of trees in the forest def find(self, x: int) -> int: """Find root of tree containing x (with path compression)""" if self.parent[x] != x: self.parent[x] = self.find(self.parent[x]) return self.parent[x] def union(self, x: int, y: int) -> bool: """ Merge trees containing x and y. Returns True if merged (were different trees). Returns False if same tree (would create cycle). """ root_x, root_y = self.find(x), self.find(y) if root_x == root_y: return False # Same tree → adding edge would create cycle # Union by rank if self.rank[root_x] < self.rank[root_y]: root_x, root_y = root_y, root_x self.parent[root_y] = root_x if self.rank[root_x] == self.rank[root_y]: self.rank[root_x] += 1 self.components -= 1 # Merged two trees into one return True def kruskal_mst(n: int, edges: List[Tuple[int, int, int]]) -> Tuple[List, int]: """ Kruskal's algorithm: Build MST by maintaining a forest. edges: List of (u, v, weight) Returns: (MST edges, total weight) The algorithm explicitly uses the forest property: - Start: Forest of n trivial trees (n components) - Each step: Add cheapest edge that doesn't create a cycle - End: Forest becomes a single tree (the MST) """ # Sort edges by weight sorted_edges = sorted(edges, key=lambda e: e[2]) uf = UnionFind(n) mst_edges = [] total_weight = 0 print(f"Starting with forest of {n} isolated vertices (trivial trees)") for u, v, weight in sorted_edges: if uf.union(u, v): mst_edges.append((u, v, weight)) total_weight += weight print(f" Added edge ({u}, {v}) weight {weight} — now {uf.components} tree(s)") if uf.components == 1: print(" Forest became a single tree (MST complete)") break else: print(f" Skipped edge ({u}, {v}) — would create cycle (same tree)") return mst_edges, total_weight # Example edges = [ (0, 1, 4), (0, 2, 3), (1, 2, 1), (1, 3, 2), (2, 3, 4), (3, 4, 2), (2, 4, 5) ] mst, weight = kruskal_mst(5, edges) print(f" MST edges: {mst}") print(f"Total MST weight: {weight}")

from collections import defaultdict, deque from typing import List, Tuple, Set def build_spanning_forest(n: int, edges: List[Tuple[int, int]]) -> List[Tuple[int, int]]: """ Build a spanning forest for a (possibly disconnected) graph. Uses BFS from each unvisited vertex to construct spanning trees for each connected component. Returns: List of edges forming the spanning forest """ adj = defaultdict(list) for u, v in edges: adj[u].append(v) adj[v].append(u) visited = set() spanning_forest_edges = [] num_trees = 0 def bfs_tree(start: int) -> List[Tuple[int, int]]: """Build spanning tree for component containing 'start'""" tree_edges = [] queue = deque([start]) visited.add(start) while queue: node = queue.popleft() for neighbor in adj[node]: if neighbor not in visited: visited.add(neighbor) tree_edges.append((node, neighbor)) queue.append(neighbor) return tree_edges for v in range(n): if v not in visited: tree_edges = bfs_tree(v) spanning_forest_edges.extend(tree_edges) num_trees += 1 print(f"Tree {num_trees}: root at {v}, edges: {tree_edges}") print(f" Spanning forest: {num_trees} trees, {len(spanning_forest_edges)} edges") print(f"Formula check: n - k = {n} - {num_trees} = {n - num_trees}") return spanning_forest_edges # Example: A disconnected graph n = 7 edges = [ (0, 1), (0, 2), (1, 2), # Triangle (component 1) (3, 4), (4, 5), # Path (component 2) # Vertex 6 is isolated (component 3) ] forest = build_spanning_forest(n, edges) print(f" Spanning forest edges: {forest}")

Forest Operations

Operation

Description

Complexity

Count components

Determine number of trees

O(V + E) via DFS/BFS

Find component

Which tree contains vertex v?

O(V + E) or O(α(n)) with Union-Find

Same component?

Are u and v in the same tree?

O(α(n)) with Union-Find

Merge trees

Connect two trees with an edge

O(α(n)) with Union-Find

Split tree

Remove an edge, creating two trees

O(V) or O(log V) with advanced structures

Component sizes

Size of each tree

O(V + E)

Total edges check

Verify forest property: |E| = n - k

O(1) given counts

from typing import List, Tuple, Optional class Forest: """ A dynamic forest data structure supporting merge and queries. Uses Union-Find for efficient component management. Tracks component sizes and validates forest invariants. """ def __init__(self, n: int): self.n = n self.parent = list(range(n)) self.rank = [0] * n self.size = [1] * n # Size of each component self.num_components = n self.num_edges = 0 def find(self, x: int) -> int: """Find root of tree containing x""" if self.parent[x] != x: self.parent[x] = self.find(self.parent[x]) return self.parent[x] def same_tree(self, u: int, v: int) -> bool: """Check if u and v are in the same tree""" return self.find(u) == self.find(v) def tree_size(self, v: int) -> int: """Get size of tree containing v""" return self.size[self.find(v)] def add_edge(self, u: int, v: int) -> bool: """ Add edge between u and v. Returns True if successful (trees merged). Returns False if would create cycle (same tree). """ root_u, root_v = self.find(u), self.find(v) if root_u == root_v: return False # Would create cycle # Union by rank if self.rank[root_u] < self.rank[root_v]: root_u, root_v = root_v, root_u self.parent[root_v] = root_u self.size[root_u] += self.size[root_v] if self.rank[root_u] == self.rank[root_v]: self.rank[root_u] += 1 self.num_components -= 1 self.num_edges += 1 return True def get_components(self) -> List[List[int]]: """Get all trees (connected components) in the forest""" components = {} for v in range(self.n): root = self.find(v) if root not in components: components[root] = [] components[root].append(v) return list(components.values()) def is_single_tree(self) -> bool: """Check if forest has become a single tree""" return self.num_components == 1 def validate(self) -> bool: """Verify forest invariant: |E| = n - k""" return self.num_edges == self.n - self.num_components def __str__(self) -> str: return (f"Forest(n={self.n}, edges={self.num_edges}, " f"components={self.num_components}, valid={self.validate()})") # Example usage forest = Forest(6) print("Initial:", forest) operations = [(0, 1), (2, 3), (4, 5), (0, 2), (3, 4)] for u, v in operations: success = forest.add_edge(u, v) status = "merged" if success else "REJECTED (cycle)" print(f"Add ({u}, {v}): {status}") print(f" {forest}") print(f" Components: {forest.get_components()}") print(f" Is single tree? {forest.is_single_tree()}")

from collections import defaultdict from typing import List, Tuple, Dict def create_rooted_forest( n: int, edges: List[Tuple[int, int]], roots: List[int] ) -> Dict[int, List[int]]: """ Root a forest at specified root vertices. Each connected component should have exactly one root. Returns children list for the rooted forest. """ adj = defaultdict(list) for u, v in edges: adj[u].append(v) adj[v].append(u) children = defaultdict(list) visited = set() def dfs_root(node: int, parent: int): """DFS to assign children based on rooting""" visited.add(node) for neighbor in adj[node]: if neighbor not in visited: children[node].append(neighbor) dfs_root(neighbor, node) # Root each component from its designated root for root in roots: if root not in visited: dfs_root(root, -1) # Handle any vertices not reached (isolated or unspecified roots) for v in range(n): if v not in visited: dfs_root(v, -1) # This vertex becomes root of its component return dict(children) def print_rooted_forest(children: Dict[int, List[int]], n: int): """Pretty print a rooted forest""" # Find roots (vertices with no parent) has_parent = set() for node, kids in children.items(): has_parent.update(kids) roots = [v for v in range(n) if v not in has_parent] def print_tree(node: int, indent: int = 0): print(" " * indent + f"└── {node}") for child in children.get(node, []): print_tree(child, indent + 1) for root in roots: print(f"Tree rooted at {root}:") print_tree(root) print() # Example: Forest with 2 trees n = 7 edges = [(0, 1), (1, 2), (1, 3), (4, 5), (5, 6)] # Tree 1: 0-1-2, 1-3 (rooted at 0) # Tree 2: 4-5-6 (rooted at 4) # Vertex 6 would be isolated if it existed without edges rooted = create_rooted_forest(n, edges, roots=[0, 4]) print("Rooted forest structure:") print_rooted_forest(rooted, n)

Forest: Multiple Disconnected Trees

When Trees Grow Together

Formal Definition of a Forest

Properties of Forests

Forests in Algorithms: Natural Occurrences

Spanning Forests: Covering Disconnected Graphs

Common Operations on Forests

Rooted Forests

Real-World Applications of Forests

There’s more to understand.

Forest: Multiple Disconnected Trees

When Trees Grow Together

Formal Definition of a Forest

Properties of Forests

Forests in Algorithms: Natural Occurrences

Spanning Forests: Covering Disconnected Graphs

Common Operations on Forests

Rooted Forests

Real-World Applications of Forests

There’s more to understand.