Count Leaves in Binary Tree Python

PROGRAM TO COUNT LEAF NODES IN A BINARY TREE

class Node:

      

    # Constructor to create a new node

    def __init__(self, data):

        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 

    else:

        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))




OUTPUT:

The leaf count of binary tree is : 3

CREDITS: https://www.geeksforgeeks.org/write-a-c-program-to-get-count-of-leaf-nodes-in-a-binary-tree/