Count Total Nodes in Binary Tree in DSA C# - Time & Space Complexity

Time Complexity: Count Total Nodes in Binary Tree

O(n)

Understanding Time Complexity

We want to understand how long it takes to count all nodes in a binary tree.

How does the time needed grow as the tree gets bigger?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

public int CountNodes(TreeNode root) {
    if (root == null) return 0;
    return 1 + CountNodes(root.left) + CountNodes(root.right);
}
    

This code counts all nodes by visiting each node once recursively.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Recursive call visiting each node once.
• How many times: Once per node in the tree.

How Execution Grows With Input

Each node is visited exactly one time, so the work grows directly with the number of nodes.

Input Size (n)	Approx. Operations
10	About 10 visits
100	About 100 visits
1000	About 1000 visits

Pattern observation: The time grows in a straight line as the tree size increases.

Final Time Complexity

Time Complexity: O(n)

This means the time to count nodes grows directly with the number of nodes in the tree.

Common Mistake

[X] Wrong: "The recursion only visits some nodes, so it is faster than O(n)."

[OK] Correct: The function must visit every node to count it, so it always touches all nodes once.

Interview Connect

Counting nodes in a tree is a basic skill that shows you understand recursion and tree traversal well.

Self-Check

"What if we changed the tree to a linked list (all nodes have only one child)? How would the time complexity change?"

Practice

1. What does the function to count total nodes in a binary tree do?

easy

A. Counts only the leaf nodes

B. Counts nodes only on the left subtree

C. Counts only the root node

D. Counts all nodes including root, left, and right children

Solution

1. Step 1: Understand the node counting concept

   The function counts every node in the tree, including root and all children.

2. Step 2: Differentiate from partial counts

   Counting only leaves or root ignores other nodes, so total count includes all nodes.

3. Final Answer:

   Counts all nodes including root, left, and right children -> Option D

4. Quick Check:

   Total nodes = all nodes in tree [OK]

Hint: Count every node visited in tree traversal [OK]

Common Mistakes:

• Counting only leaf nodes
• Ignoring root node
• Counting only one subtree

2. Which of the following is the correct C# syntax to define a binary tree node class for counting nodes?

easy

A. public class Node { public int data; public Node left, right; }

B. class Node { int data; Node left, right; void count() {} }

C. public struct Node { int data; Node left; Node right; }

D. class Node { public int data; public Node left; private Node right; }

Solution

1. Step 1: Check class and member access

   public class Node { public int data; public Node left, right; } correctly defines a public class with public members for data and child nodes.

2. Step 2: Verify syntax correctness

   class Node { int data; Node left, right; void count() {} } misses access modifiers and method is irrelevant; C uses struct which is less common; D has private right node which restricts access.

3. Final Answer:

   public class Node { public int data; public Node left, right; } -> Option A

4. Quick Check:

   Correct class with public members [OK]

Hint: Use public class with public node references [OK]

Common Mistakes:

• Missing public keyword
• Using struct instead of class
• Private child nodes blocking access

3. What is the output of the following C# code that counts nodes in a binary tree?

class Node {
  public int data;
  public Node left, right;
  public Node(int val) { data = val; left = right = null; }
}

int CountNodes(Node root) {
  if (root == null) return 0;
  return 1 + CountNodes(root.left) + CountNodes(root.right);
}

// Tree:
//     1
//    / \
//   2   3
//  / 
// 4

Node root = new Node(1);
root.left = new Node(2);
root.right = new Node(3);
root.left.left = new Node(4);

Console.WriteLine(CountNodes(root));

medium

A. 4

B. 3

C. 5

D. 6

Solution

1. Step 1: Count nodes in the tree structure

   The tree has nodes: 1 (root), 2 (left child), 3 (right child), and 4 (left child of 2), total 4 nodes.

2. Step 2: Apply recursive count function

   CountNodes returns 1 + left subtree count + right subtree count, summing to 4.

3. Final Answer:

   4 -> Option A

4. Quick Check:

   Nodes counted = 4 [OK]

Hint: Add 1 for root plus counts of left and right [OK]

Common Mistakes:

• Counting only children, missing root
• Miscounting null nodes
• Adding extra nodes

4. Identify the error in this C# method to count total nodes in a binary tree:

int CountNodes(Node root) {
  if (root == null) return 1;
  return CountNodes(root.left) + CountNodes(root.right);
}

medium

A. Returns 1 for null node instead of 0

B. Does not count the current root node

C. All of the above

D. Missing addition of 1 for current node

Solution

1. Step 1: Check base case return value

   Returning 1 for null node is incorrect; it should be 0 because null means no node.

2. Step 2: Check counting logic for current node

   Function does not add 1 for the current root node, so it misses counting it.

3. Step 3: Combine errors

   Both base case and missing addition cause wrong total count.

4. Final Answer:

   All of the above -> Option C

5. Quick Check:

   Base case and counting errors combined [OK]

Hint: Base case returns 0; add 1 for current node [OK]

Common Mistakes:

• Returning 1 for null node
• Not adding 1 for current node
• Ignoring base case correctness

5. Given a binary tree where some nodes have only one child, how would you modify the counting function to count only nodes that have at least one child?

hard

A. Count only leaf nodes

B. Count nodes where left or right child is not null, recursively

C. Count all nodes including leaves

D. Count nodes with both children null

Solution

1. Step 1: Define condition for counting nodes

   We want to count nodes that have at least one child, so check if left or right child is not null.

2. Step 2: Modify recursive function accordingly

   Recursively count nodes where left or right child exists, ignoring leaf nodes with no children.

3. Step 3: Implement logic

   Return 1 + counts of children if node has child; else 0 for leaf nodes.

4. Final Answer:

   Count nodes where left or right child is not null, recursively -> Option B

5. Quick Check:

   Count nodes with children only [OK]

Hint: Count nodes with at least one child only [OK]

Common Mistakes:

• Counting all nodes including leaves
• Counting only leaves
• Counting nodes with no children