P02_CountLeafNodes.py

# Author: OMKAR PATHAK

# leaf node is the one which does not have any children

from Tree import Node

def countLeafNodes(root):
    if root is None:
        return 0
    if(root.left is None and root.right is None):
        return 1
    else:
        return countLeafNodes(root.left) + countLeafNodes(root.right)

if __name__ == '__main__':
    root = Node(1)
    root.setLeft(Node(2))
    root.setRight(Node(3))
    root.left.setLeft(Node(4))

    print('Count of leaf nodes:',countLeafNodes(root))