Lowest Common Ancestor in Binary Tree in DSA C++ - Build from Scratch

Lowest Common Ancestor in Binary Tree

📖 Scenario: You are working on a family tree application. You want to find the closest common ancestor of two family members in the tree.

🎯 Goal: Build a program to find the Lowest Common Ancestor (LCA) of two nodes in a binary tree.

📋 What You'll Learn

Create a binary tree with specific nodes

Add variables to represent the two nodes to find the LCA for

Implement a function to find the LCA of the two nodes

Print the value of the LCA node

💡 Why This Matters

🌍 Real World

Finding common ancestors is useful in family tree applications, organizational charts, and network routing.

💼 Career

Understanding tree traversal and recursive algorithms is important for software engineering roles involving data structures and algorithms.

1

Create the Binary Tree Structure

Create a struct called TreeNode with an int val, and two pointers left and right. Then create the binary tree with these exact nodes and connections:
1 as root, 2 as root's left child, 3 as root's right child, 4 as left child of 2, and 5 as right child of 2.

DSA C++

// Define TreeNode struct and create the binary tree nodes
// Your code here

Hint

Start by defining the TreeNode struct with a constructor. Then create nodes and link them as described.

2

Set the Nodes to Find LCA For

Add two pointers called p and q and set them to point to the nodes with values 4 and 5 respectively in the tree.

DSA C++

#include 

struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

int main() {
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    // Set pointers p and q to nodes 4 and 5
    // Your code here

    return 0;
}

Hint

Use the tree structure to assign p and q to the correct nodes.

3

Implement the Lowest Common Ancestor Function

Write a function called lowestCommonAncestor that takes TreeNode* root, TreeNode* p, and TreeNode* q as parameters and returns the pointer to their lowest common ancestor node. Use recursion to check if root is nullptr, or equals p or q. Recursively search left and right subtrees, and return the correct ancestor node.

DSA C++

#include 

struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
    // Implement the recursive LCA logic here
    // Your code here
}

int main() {
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    TreeNode* p = root->left->left;  // Node with value 4
    TreeNode* q = root->left->right; // Node with value 5

    return 0;
}

Hint

Use recursion to find p and q in left and right subtrees. If both sides return non-null, current node is LCA.

4

Print the Lowest Common Ancestor Value

Call the lowestCommonAncestor function with root, p, and q. Store the result in a variable called ancestor. Then print the value of ancestor->val using std::cout.

DSA C++

#include 

struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
    if (root == nullptr || root == p || root == q) {
        return root;
    }
    TreeNode* left = lowestCommonAncestor(root->left, p, q);
    TreeNode* right = lowestCommonAncestor(root->right, p, q);
    if (left != nullptr && right != nullptr) {
        return root;
    }
    return left != nullptr ? left : right;
}

int main() {
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    TreeNode* p = root->left->left;  // Node with value 4
    TreeNode* q = root->left->right; // Node with value 5

    // Call lowestCommonAncestor and print the result
    // Your code here

    return 0;
}

Hint

Call the function and print the val of the returned node.

Practice

1. What is the Lowest Common Ancestor (LCA) of two nodes in a binary tree?

easy

A. The deepest node that is an ancestor of both nodes

B. The root node of the binary tree

C. The node with the smallest value between the two nodes

D. The node that appears first in an inorder traversal

Solution

1. Step 1: Understand the definition of LCA

   The Lowest Common Ancestor is the deepest node in the tree that has both given nodes as descendants (a node can be a descendant of itself).

2. Step 2: Differentiate from other nodes

   It is not necessarily the root or the node with smallest value, but the common ancestor closest to the two nodes.

3. Final Answer:

   The deepest node that is an ancestor of both nodes -> Option A

4. Quick Check:

   LCA = deepest common ancestor [OK]

Hint: LCA is the lowest node common to both paths [OK]

Common Mistakes:

• Confusing LCA with root node
• Choosing node by value instead of position
• Thinking LCA is the first visited node in traversal

2. Which of the following is the correct C++ function signature to find LCA in a binary tree node structure?

easy

A. int findLCA(int root, int n1, int n2);

B. TreeNode* findLCA(TreeNode* root, int n1, int n2);

C. void findLCA(TreeNode root, int n1, int n2);

D. TreeNode findLCA(TreeNode root, int n1, int n2);

Solution

1. Step 1: Analyze parameter types

   The function should accept a pointer to the root node and two integer values representing node keys.

2. Step 2: Analyze return type

   The function returns a pointer to the TreeNode representing the LCA, so return type must be TreeNode*.

3. Final Answer:

   TreeNode* findLCA(TreeNode* root, int n1, int n2); -> Option B

4. Quick Check:

   Function signature with pointers and correct return type [OK]

Hint: LCA function returns pointer to TreeNode [OK]

Common Mistakes:

• Using value instead of pointer for root
• Returning int instead of TreeNode*
• Passing root by value causing copy

3. Given the following binary tree and code snippet, what will be the output?

#include <iostream>
struct TreeNode {
  int val;
  TreeNode* left;
  TreeNode* right;
  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

TreeNode* findLCA(TreeNode* root, int n1, int n2) {
  if (!root) return nullptr;
  if (root->val == n1 || root->val == n2) return root;
  TreeNode* left = findLCA(root->left, n1, n2);
  TreeNode* right = findLCA(root->right, n1, n2);
  if (left && right) return root;
  return left ? left : right;
}

int main() {
  TreeNode* root = new TreeNode(3);
  root->left = new TreeNode(5);
  root->right = new TreeNode(1);
  root->left->left = new TreeNode(6);
  root->left->right = new TreeNode(2);
  root->right->left = new TreeNode(0);
  root->right->right = new TreeNode(8);
  TreeNode* lca = findLCA(root, 6, 2);
  std::cout << lca->val << std::endl;
  return 0;
}

medium

A. 6

B. 3

C. 2

D. 5

Solution

1. Step 1: Trace the tree structure and nodes

   Nodes 6 and 2 are both children of node 5 in the tree.

2. Step 2: Apply LCA logic

   Since 5 is the lowest node that has both 6 and 2 as descendants, the function returns node 5.

3. Final Answer:

   5 -> Option D

4. Quick Check:

   LCA(6,2) = 5 [OK]

Hint: LCA is ancestor closest to both nodes [OK]

Common Mistakes:

• Choosing root 3 instead of node 5
• Confusing child nodes as LCA
• Returning one of the input nodes incorrectly

4. Identify the bug in the following LCA function and select the correct fix:

TreeNode* findLCA(TreeNode* root, int n1, int n2) {
  if (!root) return nullptr;
  if (root->val == n1 || root->val == n2) return root;
  TreeNode* left = findLCA(root->left, n1, n2);
  TreeNode* right = findLCA(root->right, n1, n2);
  if (left && right) return root;
  if (left) return right;
  return right;
}

medium

A. Add a check for root being nullptr after recursive calls

B. Change 'if (left && right) return root;' to 'return nullptr;'

C. Change 'if (left) return right;' to 'if (left) return left;'

D. Change return type to int instead of TreeNode*

Solution

1. Step 1: Analyze the return statements

   The code incorrectly returns 'right' when 'left' is true, which is wrong logic.

2. Step 2: Correct the return for left subtree

   It should return 'left' when 'left' is true and 'right' is false to propagate the found node correctly.

3. Final Answer:

   Change 'if (left) return right;' to 'if (left) return left;' -> Option C

4. Quick Check:

   Return correct subtree pointer [OK]

Hint: Return the non-null subtree pointer correctly [OK]

Common Mistakes:

• Returning wrong subtree pointer
• Ignoring nullptr checks
• Changing return type unnecessarily

5. Given a binary tree where nodes may not contain unique values, which modification is necessary to correctly find the LCA of two nodes identified by their pointers (not values)?

hard

A. Modify the LCA function to compare node pointers directly instead of values

B. Use node values for comparison as usual, ignoring pointer identity

C. Sort the tree nodes before finding LCA to ensure uniqueness

D. Replace the binary tree with a binary search tree for unique values

Solution

1. Step 1: Understand the problem with duplicate values

   If nodes have duplicate values, comparing values is ambiguous and can lead to wrong LCA.

2. Step 2: Use pointer comparison

   Comparing node pointers directly ensures the exact nodes are identified regardless of value duplicates.

3. Step 3: Modify function accordingly

   The LCA function should check if the current node pointer equals either of the target node pointers.

4. Final Answer:

   Modify the LCA function to compare node pointers directly instead of values -> Option A

5. Quick Check:

   Compare pointers, not values, for duplicates [OK]

Hint: Compare node addresses, not values, for duplicates [OK]

Common Mistakes:

• Using values for comparison with duplicates
• Assuming tree is BST to fix duplicates
• Sorting nodes which is unnecessary