Path Compression in Union Find in DSA Go - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is the main purpose of path compression in Union Find?

Path compression helps speed up the 'find' operation by making nodes on the path point directly to the root, flattening the structure for faster future queries.

intermediate

Explain how path compression changes the parent pointers during a find operation.

During find, as we move up from a node to its root, path compression updates each node's parent pointer to point directly to the root, reducing the tree height.

intermediate

Why does path compression improve the efficiency of Union Find operations?

Because it flattens the tree structure, future 'find' operations take almost constant time, making the overall algorithm very efficient for many union/find operations.

advanced

Show the effect of path compression on the parent array after calling find on node 4 in this structure: parent = [0, 0, 1, 2, 3]

After find(4) with path compression, parent array becomes [0, 0, 0, 0, 0] because nodes 1, 2, 3, and 4 on the path are updated to point directly to root 0.

advanced

What is the time complexity of Union Find with path compression for m operations on n elements?

The amortized time complexity is almost O(m α(n)), where α(n) is the inverse Ackermann function, which grows very slowly and is practically constant.

What does path compression do in the Union Find data structure?

A Makes every node point directly to the root during find

B Merges two sets by size

C Deletes nodes from the structure

D Splits sets into smaller parts

Path compression updates nodes to point directly to the root during find, flattening the tree.

When is path compression applied in Union Find?

A During union operation

B During find operation

C When initializing the structure

D When deleting elements

Path compression is applied during the find operation to speed up future finds.

What is the effect of path compression on the height of the tree?

A Decreases height

B Keeps height the same

C Increases height

D Removes the tree

Path compression decreases the height by making nodes point directly to the root.

Which function describes the amortized time complexity of Union Find with path compression?

A O(n)

B O(log n)

C O(α(n))

D O(n^2)

The inverse Ackermann function α(n) describes the almost constant amortized time complexity.

What happens to the parent pointer of a node after path compression?

A It becomes null

B It points to its immediate parent

C It points to a random node

D It points to the root of the set

After path compression, the node's parent pointer points directly to the root.

Describe how path compression works in the Union Find data structure and why it improves performance.

Think about what happens when you look for the root of a node.

Explain the time complexity impact of path compression on multiple union and find operations.

Consider how path compression affects repeated finds.

Practice

1. What is the main purpose of path compression in the Union Find data structure?

easy

A. To speed up the find operation by making nodes point directly to the root

B. To merge two sets by size

C. To initialize the parent array

D. To count the number of connected components

Solution

1. Step 1: Understand the role of path compression

   Path compression is used during the find operation to flatten the tree structure by making nodes point directly to the root.

2. Step 2: Compare with other operations

   Merging sets by size is done in union by size/rank, not path compression. Initialization and counting components are unrelated to path compression.

3. Final Answer:

   To speed up the find operation by making nodes point directly to the root -> Option A

4. Quick Check:

   Path compression optimizes find = To speed up the find operation by making nodes point directly to the root [OK]

Hint: Path compression flattens trees to speed up find [OK]

Common Mistakes:

• Confusing path compression with union by size/rank
• Thinking it initializes the data structure
• Believing it counts components

2. Which of the following is the correct Go syntax for the find function with path compression in Union Find?

easy

A. func find(parent []int, x int) int { for parent[x] != x { x = parent[x] } return x }

B. func find(parent []int, x int) int { parent[x] = x; return x }

C. func find(parent []int, x int) int { if parent[x] == x { return x } else { return find(parent, x) } }

D. func find(parent []int, x int) int { if parent[x] != x { parent[x] = find(parent, parent[x]) } return parent[x] }

Solution

1. Step 1: Identify path compression in the code

   func find(parent []int, x int) int { if parent[x] != x { parent[x] = find(parent, parent[x]) } return parent[x] } recursively updates parent[x] to the root, which is path compression.

2. Step 2: Check other options for correctness

   func find(parent []int, x int) int { for parent[x] != x { x = parent[x] } return x } uses a loop but no path compression. func find(parent []int, x int) int { if parent[x] == x { return x } else { return find(parent, x) } } calls find without updating parent[x]. func find(parent []int, x int) int { parent[x] = x; return x } wrongly resets parent[x] to itself always.

3. Final Answer:

   func find(parent []int, x int) int { if parent[x] != x { parent[x] = find(parent, parent[x]) } return parent[x] } -> Option D

4. Quick Check:

   Recursive update of parent[x] = find(...) = func find(parent []int, x int) int { if parent[x] != x { parent[x] = find(parent, parent[x]) } return parent[x] } [OK]

Hint: Look for recursive parent update for path compression [OK]

Common Mistakes:

• Missing parent update in recursion
• Using loop without compression
• Resetting parent incorrectly

3. What will be the output of the following Go code snippet using Union Find with path compression?

package main
import "fmt"

func find(parent []int, x int) int {
    if parent[x] != x {
        parent[x] = find(parent, parent[x])
    }
    return parent[x]
}

func main() {
    parent := []int{0, 1, 2, 3, 4}
    parent[1] = 0
    parent[2] = 1
    fmt.Println(find(parent, 2))
    fmt.Println(parent)
}

medium

A. 1
[0 0 1 3 4]

B. 0
[0 0 0 3 4]

C. 2
[0 1 2 3 4]

D. 0
[0 1 0 3 4]

Solution

1. Step 1: Trace the find call for element 2

   find(2) calls find(1), which calls find(0). Since parent[0] = 0, root is 0.

2. Step 2: Apply path compression updates

   parent[2] is updated to 0, parent[1] is updated to 0. Final parent array is [0 0 0 3 4].

3. Final Answer:

   0 [0 0 0 3 4] -> Option B

4. Quick Check:

   Root of 2 is 0 and parent array updated = 0 [0 0 0 3 4] [OK]

Hint: Follow recursion and update parents on return [OK]

Common Mistakes:

• Not updating parent array after find
• Returning intermediate parent instead of root
• Confusing indices in parent array

4. Identify the bug in the following Go code implementing find with path compression:

func find(parent []int, x int) int {
    if parent[x] == x {
        return x
    }
    parent[x] = find(parent, x)
    return parent[x]
}

medium

A. The function should not update parent[x] during recursion

B. The base case should check parent[x] != x instead of parent[x] == x

C. The recursive call should be find(parent, parent[x]) instead of find(parent, x)

D. The function should return x always

Solution

1. Step 1: Analyze the recursive call argument

   The recursive call uses find(parent, x), which causes infinite recursion because x does not change.

2. Step 2: Correct the recursive call

   It should call find(parent, parent[x]) to move up the tree towards the root.

3. Final Answer:

   The recursive call should be find(parent, parent[x]) instead of find(parent, x) -> Option C

4. Quick Check:

   Recursive call must move up tree = The recursive call should be find(parent, parent[x]) instead of find(parent, x) [OK]

Hint: Recursive find must use parent[x], not x [OK]

Common Mistakes:

• Calling find with same x causing infinite recursion
• Wrong base case condition
• Not updating parent[x] for compression

5. Given a Union Find structure with parent = [0, 0, 1, 2, 3], what will be the parent array after calling find(parent, 4) with path compression?

hard

A. [0, 0, 0, 0, 0]

B. [0, 0, 1, 2, 0]

C. [0, 1, 1, 2, 3]

D. [0, 0, 1, 3, 4]

Solution

1. Step 1: Trace the find call for element 4

   find(4) calls find(3), find(3) calls find(2), find(2) calls find(1), find(1) calls find(0). Root is 0.

2. Step 2: Apply path compression updates

   During recursion unwind: parent[1] = 0 (already), parent[2] = 0, parent[3] = 0, parent[4] = 0. Final parent array is [0, 0, 0, 0, 0].

3. Final Answer:

   [0, 0, 0, 0, 0] -> Option A

4. Quick Check:

   All nodes on path updated to root 0 = [0, 0, 0, 0, 0] [OK]

Hint: Update only nodes on path to root during find [OK]

Common Mistakes:

• Only updating parent[4] to root
• Failing to compress the full path
• Confusing parent indices