Diameter of Binary Tree in DSA C++ - Interactive Practice

Practice - 5 Tasks

1fill in blank

easy

Complete the code to calculate the height of a binary tree node.

DSA C++

int height(TreeNode* node) {
    if (node == nullptr) return 0;
    int leftHeight = height(node->left);
    int rightHeight = height(node->right);
    return 1 + [1];
}

Drag options to blanks, or click blank then click option'

AleftHeight - rightHeight

Bmin(leftHeight, rightHeight)

CleftHeight + rightHeight

Dmax(leftHeight, rightHeight)

Attempts:

3 left

💡 Hint

Common Mistakes

Using min instead of max for height calculation.

Adding left and right heights instead of taking max.

The height of a node is 1 plus the maximum height of its left and right subtrees.

2fill in blank

medium

Complete the code to update the diameter while computing height.

DSA C++

int diameter = 0;

int height(TreeNode* node) {
    if (node == nullptr) return 0;
    int leftHeight = height(node->left);
    int rightHeight = height(node->right);
    diameter = std::max(diameter, leftHeight + rightHeight + [1]);
    return 1 + std::max(leftHeight, rightHeight);
}

Drag options to blanks, or click blank then click option'

A0

B-1

C1

D2

Attempts:

3 left

💡 Hint

Common Mistakes

Adding 0 instead of 1, missing the current node.

Adding 2 or -1 causing wrong diameter count.

We add 1 to count the current node between left and right paths for diameter.

3fill in blank

hard

Fix the error in the diameter calculation function to return the correct diameter.

DSA C++

int diameterOfBinaryTree(TreeNode* root) {
    diameter = 0;
    height(root);
    return [1];
}

Drag options to blanks, or click blank then click option'

Adiameter

Bheight(root)

C0

Droot->val

Attempts:

3 left

💡 Hint

Common Mistakes

Returning height instead of diameter.

Returning 0 or root value instead of diameter.

The diameter variable stores the maximum diameter found, so return it.

4fill in blank

hard

Fill both blanks to complete the recursive height function that updates diameter.

DSA C++

int height(TreeNode* node) {
    if (node == nullptr) return 0;
    int leftHeight = height(node->[1]);
    int rightHeight = height(node->[2]);
    diameter = std::max(diameter, leftHeight + rightHeight + 1);
    return 1 + std::max(leftHeight, rightHeight);
}

Drag options to blanks, or click blank then click option'

Aleft

Bright

Cparent

Dchild

Attempts:

3 left

💡 Hint

Common Mistakes

Using parent or child which are not valid pointers here.

Swapping left and right causing confusion.

We recursively call height on the left and right child nodes.

5fill in blank

hard

Fill all three blanks to complete the diameter calculation using a helper function.

DSA C++

class Solution {
public:
    int diameter = 0;

    int height(TreeNode* [1]) {
        if ([2] == nullptr) return 0;
        int leftHeight = height([2]->left);
        int rightHeight = height([2]->right);
        diameter = std::max(diameter, leftHeight + rightHeight + 1);
        return 1 + std::max(leftHeight, rightHeight);
    }

    int diameterOfBinaryTree(TreeNode* root) {
        height([3]);
        return diameter - 1;
    }
};

Drag options to blanks, or click blank then click option'

Anode

Croot

Dthis

Attempts:

3 left

💡 Hint

Common Mistakes

Using inconsistent variable names causing errors.

Calling height with 'this' or wrong variable.

The helper function uses 'node' as parameter and variable; diameterOfBinaryTree calls height with root.

Practice

1. What does the diameter of a binary tree represent?

easy

A. The longest path between any two nodes in the tree

B. The height of the tree

C. The number of leaf nodes

D. The total number of nodes in the tree

Solution

1. Step 1: Understand the diameter definition

   The diameter is the longest path between any two nodes, not just from root to leaf.

2. Step 2: Differentiate from height and node count

   Height is longest root-to-leaf path; node count and leaf count are unrelated to diameter.

3. Final Answer:

   The longest path between any two nodes in the tree -> Option A

4. Quick Check:

   Diameter = longest node-to-node path [OK]

Hint: Diameter is longest path between any two nodes [OK]

Common Mistakes:

• Confusing diameter with tree height
• Thinking diameter counts nodes instead of edges
• Assuming diameter is number of leaf nodes

2. Which of the following is the correct C++ function signature to compute the diameter of a binary tree node?

easy

A. int diameter(TreeNode& root);

B. void diameter(TreeNode root);

C. int diameter(TreeNode root);

D. int diameter(TreeNode* root);

Solution

1. Step 1: Understand pointer usage in tree nodes

   Binary trees are usually handled with pointers to nodes, so TreeNode* is correct.

2. Step 2: Check return type and parameter correctness

   Diameter returns an int, and parameter should be pointer, not value or reference for typical tree traversal.

3. Final Answer:

   int diameter(TreeNode* root); -> Option D

4. Quick Check:

   Pointer parameter with int return [OK]

Hint: Use pointer parameter and int return type [OK]

Common Mistakes:

• Passing node by value instead of pointer
• Using void return type instead of int
• Passing node by reference unnecessarily

3. What is the output of the following code snippet computing diameter?

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

int height(TreeNode* root) {
  if (!root) return 0;
  return 1 + std::max(height(root->left), height(root->right));
}

int diameter(TreeNode* root) {
  if (!root) return 0;
  int leftHeight = height(root->left);
  int rightHeight = height(root->right);
  int leftDiameter = diameter(root->left);
  int rightDiameter = diameter(root->right);
  return std::max(leftHeight + rightHeight, std::max(leftDiameter, rightDiameter));
}

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);
  std::cout << diameter(root);
  return 0;
}

medium

A. 6

B. 4

C. 3

D. 5

Solution

1. Step 1: Calculate heights of left and right subtrees

   Left subtree height: max height of nodes 2->4 or 2->5 is 2; right subtree height (node 3) is 1.

2. Step 2: Calculate diameter candidates and choose max

   Diameter candidates: leftHeight + rightHeight = 2 + 1 = 3; leftDiameter and rightDiameter are computed recursively. LeftDiameter is 2 (path 4-2-5), rightDiameter is 0. Max is 3.

3. Final Answer:

   3 -> Option C

4. Quick Check:

   Longest path length = 3 edges [OK]

Hint: Sum left and right subtree heights for diameter [OK]

Common Mistakes:

• Confusing height with diameter
• Counting nodes instead of edges
• Ignoring recursive diameter calls

4. Identify the bug in this diameter function and select the fix:

int diameter(TreeNode* root) {
  if (!root) return 0;
  int leftHeight = height(root->left);
  int rightHeight = height(root->right);
  int leftDiameter = diameter(root->left);
  int rightDiameter = diameter(root->right);
  return leftHeight + rightHeight;
}

medium

A. Return max of leftDiameter and rightDiameter only

B. Return max of leftHeight + rightHeight and leftDiameter, rightDiameter

C. Return sum of leftDiameter and rightDiameter

D. Return height of root node

Solution

1. Step 1: Analyze current return statement

   Currently returns only leftHeight + rightHeight, ignoring recursive diameters from subtrees.

2. Step 2: Correct return to consider all diameter candidates

   Diameter is max of leftHeight + rightHeight, leftDiameter, and rightDiameter.

3. Final Answer:

   Return max of leftHeight + rightHeight and leftDiameter, rightDiameter -> Option B

4. Quick Check:

   Include all diameter candidates [OK]

Hint: Return max of height sum and subtree diameters [OK]

Common Mistakes:

• Ignoring recursive diameter calls
• Returning sum of diameters incorrectly
• Returning height instead of diameter

5. Given a binary tree where each node has a value, how would you modify the diameter function to return the maximum sum of node values on any path between two nodes instead of the number of edges?

hard

A. Change height function to return max sum of node values and update diameter to use sums instead of heights

B. Use the same diameter function but multiply the result by node values

C. Calculate diameter normally and add all node values in the tree

D. Return the maximum node value in the tree as diameter

Solution

1. Step 1: Modify height to return max sum of node values

   Instead of counting edges, height returns max sum from node to leaf including node values.

2. Step 2: Update diameter to use sums from left and right subtrees

   Diameter becomes max of left sum + right sum and recursive diameters, reflecting max path sum.

3. Final Answer:

   Change height function to return max sum of node values and update diameter to use sums instead of heights -> Option A

4. Quick Check:

   Use sums in height and diameter for max path value [OK]

Hint: Use sums in height and diameter for max path value [OK]

Common Mistakes:

• Multiplying diameter by node values incorrectly
• Adding all node values regardless of path
• Returning max node value instead of path sum