Code : Count leaf nodes

 

Given a generic tree, count and return the number of leaf nodes present in the given tree.

Input format :

The first line of input contains data of the nodes of the tree in level order form. The order is: data for root node, number of children to root node, data of each of child nodes and so on and so forth for each node. The data of the nodes of the tree is separated by space.  

Output Format :

The first and only line of output prints the count of leaf nodes present in the given tree.

Constraints:

Time Limit: 1 sec

Sample Input 1 :

10 3 20 30 40 2 40 50 0 0 0 0 

Sample Output 1 :

4

/************************************************************
 
    Following is the structure for the TreeNode class
    template <typename T>
    class TreeNode {
     public:
        T data;
        vector<TreeNode<T>*> children;
    
        TreeNode(T data) {
            this->data = data;
        }
    
        ~TreeNode() {
            for (int i = 0; i < children.size(); i++) {
                delete children[i];
            }
        }
    };
************************************************************/
int getLeafNodeCount(TreeNode<int>* root) {
    // Write your code here
    if(root==NULL)return NULL;
    if(root->children.size()==0) return 1;
    int count=0;
    for(int i=0;i<root->children.size();i++){
        count+=getLeafNodeCount(root->children[i]);
    }
    return count;
    
    
    
    
//     if(root==NULL)return NULL;
//     if(root->children.size()==0) return 1;
//     int count=0;
//     for(int i=0;i<root->children.size();i++){
//       count+= getLeafNodeCount(root->children[i]);
       
//     }
//     return count;
    
}