Check if a Binary Tree is Complete Binary Tree
June 25, 2013
Linear Search in detail
June 29, 2013

Check if all leaf nodes are at the same level

watch_videoGiven a binary tree write a function to check if all leaf nodes of the tree are at the same level. For example, In all the below trees, the leaf nodes are at same level:
single treebinarytree_2
complete binary tree
binarytree_1
But in the below tree the leaf nodes are not at the same level:

   A           A              A
  / \         / \           /   \
 B   C       B   C         B     C
    /         \           / \
   D           D         D  E

Method-1: Calculate distance from the root:

This is similar to saying, that check if all leaf nodes are at the same distance from the root. This can be checked by making small changes in the BFS code to do level wise traversal.

Basically we have to check that, (the first leaf node is seen at the start of a level) AND (All the nodes after the first leaf node should also be leaf nodes)

Method-2:

Algorithm

This algorithm keeps track of the level of current node. It also has a static variable which will be set with the level of leaf node, when we encounter a leaf node for the first time. Every time we see a leaf node we check whether the level of this leaf node is same as the one set by us.. If no then this leaf node is at different level from the one found previously. Hence, return false, else continue till all the nodes are seen.

    1. Reset global variable, leafLevel = -1
    2. Traverse the entire tree. For each node:
    3. IF(current node is a leaf node)
          IF(leafLevel == -1)              // Not yet set
             leafLevel = level of Current Node
             RETURN true                   // Case when there is just 1 node
          ELSE
             IF(leafLevel == level of Current Node)
                 RETURN true;
             ELSE
                 RETURN false;
        ESLE
            call function recursively for left and right subtree and
                 return true if both of them returns true, else return false

Code:

/**
 * This function should not be called directly because it does not handle the case where r is null.
 * Either handle the case at the level of caller or call the next function (checkAllNodesSameLevel)
 */
int checkAllNodesSameLvlRecursive(struct node * r, int curLevel, int resetTopLevel)
{
    static int topLevel = -1;
    if(resetTopLevel)
        topLevel = -1;
    // If it is a leaf node, then take action.
    if(r->lptr == NULL && r->rptr == NULL)
    {
        // If it is the first leaf node encountered
        if(topLevel == -1)
        {
            topLevel = curLevel;
            return true;
        }
        else if(curLevel == topLevel)
            return true;
        else
            return false;
    }
    // Check the left and right subtree.
    int leftResult = true;
    int rightResult = true;
    if(r->lptr)
        leftResult = checkAllNodesSameLvlRecursive(r->lptr, curLevel+1, false);
    if(r->rptr)
        rightResult = checkAllNodesSameLvlRecursive(r->rptr, curLevel+1, false);
    if( !rightResult || !leftResult)
        return false;
    return true;
}
/**
 * Main function to be called from outside to check
 * if all leaf nodes are at the same level.
 */
int checkAllNodesSameLevel(struct node * r)
{
    if(r == NULL)
        return true;
    return checkAllNodesSameLvlRecursive(r, 0, true);
}

Time Complexity: O(n), Since we are visiting a node only once.

2 Comments

  1. It’s awesome designed for me to have a website, which is good for my know-how.
    thanks admin

  2. Dieter says:

    This is the right blog for anybody who really wants to understand this topic.
    You know so much its almost tough to argue with you (not that I really would want to…HaHa).
    You certainly put a brand new spin on a subject that
    has been written about for years. Wonderful stuff, just excellent!

Leave a Reply to Anonymous Cancel reply

Your email address will not be published. Required fields are marked *