Program to count leaf nodes in a binary tree

A node is a leaf node if both left and right child nodes of it are NULL.

Here is an algorithm to get the leaf node count.

1) If node is NULL then return 0.
2) Else If left and right child nodes are NULL return 1.
3) Else recursively calculate leaf count of the tree using below formula.
    Leaf count of a tree = Leaf count of left subtree + 
                                 Leaf count of right subtree

Example Tree

Example Tree

Leaf count for the above tree is 3.



#include <stdio.h>
#include <stdlib.h>

/* A binary tree node has data, pointer to left child 
   and a pointer to right child */
struct node 
    int data;
    struct node* left;
    struct node* right;

/* Function to get the count of leaf nodes in a binary tree*/
unsigned int getLeafCount(struct node* node)
  if(node == NULL)       
    return 0;
  if(node->left == NULL && node->right==NULL)      
    return 1;            
    return getLeafCount(node->left)+

/* Helper function that allocates a new node with the
   given data and NULL left and right pointers. */
struct node* newNode(int data) 
  struct node* node = (struct node*)
                       malloc(sizeof(struct node));
  node->data = data;
  node->left = NULL;
  node->right = NULL;

/*Driver program to test above functions*/  
int main()
  /*create a tree*/  
  struct node *root = newNode(1);
  root->left        = newNode(2);
  root->right       = newNode(3);
  root->left->left  = newNode(4);
  root->left->right = newNode(5);    
  /*get leaf count of the above created tree*/
  printf("Leaf count of the tree is %d", getLeafCount(root));
  return 0;


//Java implementation to find leaf count of a given Binary tree
/* Class containing left and right child of current 
 node and key value*/
class Node 
    int data;
    Node left, right;
    public Node(int item) 
        data = item;
        left = right = null;
public class BinaryTree 
    //Root of the Binary Tree
    Node root;
    /* Function to get the count of leaf nodes in a binary tree*/
    int getLeafCount() 
        return getLeafCount(root);
    int getLeafCount(Node node) 
        if (node == null)
            return 0;
        if (node.left == null && node.right == null)
            return 1;
            return getLeafCount(node.left) + getLeafCount(node.right);
    /* Driver program to test above functions */
    public static void main(String args[]) 
        /* create a tree */
        BinaryTree tree = new BinaryTree();
        tree.root = new Node(1);
        tree.root.left = new Node(2);
        tree.root.right = new Node(3);
        tree.root.left.left = new Node(4);
        tree.root.left.right = new Node(5);
        /* get leaf count of the abve tree */
        System.out.println("The leaf count of binary tree is : " 
                             + tree.getLeafCount());

// This code has been contributed by Mayank Jaiswal(mayank_24)


# Python program to count leaf nodes in Binary Tree

# A Binary tree node
class Node:
    # Constructor to create a new node
    def __init__(self, data): = data 
        self.left = None
        self.right = None

# Function to get the count of leaf nodes in binary tree
def getLeafCount(node):
    if node is None:
        return 0 
    if(node.left is None and node.right is None):
        return 1 
        return getLeafCount(node.left) + getLeafCount(node.right)

# Driver program to test above function
root = Node(1)
root.left = Node(2)
root.right = Node(3)
root.left.left = Node(4)
root.left.right = Node(5)

print "Leaf count of the tree is %d" %(getLeafCount(root))

#This code is contributed by Nikhil Kumar Singh(nickzuck_007)

Time & Space Complexities: Since this program is similar to traversal of tree, time and space complexities will be same as Tree traversal (Please see our Tree Traversal post for details)

Please write comments if you find any bug in the above programs/algorithms or other ways to solve the same problem.

Please write to us at to report any issue with the above content.

Recommended Posts:

1.6 Average Difficulty : 1.6/5.0
Based on 167 vote(s)