Diameter of Binary Tree

PROBLEM :

Given a Binary Tree, find diameter of it.  The diameter of a tree is the number of nodes on the longest path between two leaves in the tree. The diagram below shows two trees each with diameter nine, the leaves that form the ends of a longest path are shaded (note that there is more than one path in each tree of length nine, but no path longer than nine nodes).




Input:
The task is to complete the method which takes one argument, root of Binary Tree. The struct Node has a data part which stores the data, pointer to left child and pointer to right child.
There are multiple test cases. For each test case, this method will be called individually.

Output:
The function should return diameter of binary tree.

Constraints:
1 <=T<= 30
1 <=Number of nodes<= 100
1 <=Data of a node<= 1000

Example:
Input:
2
2
1 2 L 1 3 R
4
10 20 L 10 30 R 20 40 L 20 60 R

Output:
3
4

There are two test casess.  First case represents a tree with 3 nodes and 2 edges where root is 1, left child of 1 is 2 and right child of 1 is 3.   Second test case represents a tree with 4 edges and 5 nodes

--------------------------------------------------------------------------------
SIMPLE c++ IMPLEMENTATION :
--------------------------------------------------------------------------------

/* Tree node structure  used in the program
 struct Node
 {
     int data;
     struct Node* left, *right;
}; */

/* Computes the diameter of binary tree with given root.  */

int height(struct Node *) ;
int max(int ,int ) ;

int diameter(struct Node* root)
{
   if(root==NULL)
        return 0 ;
     
    int Lheight,Rheight ;
    int Ldia,Rdia ;
 
    Lheight=height(root->left) ;
    Rheight=height(root->right) ;
 
    Ldia=diameter(root->left) ;
    Rdia=diameter(root->right) ;
 
    return max(Lheight+Rheight+1,max(Ldia,Rdia)) ;
}

int height(struct Node *root)
{
    if(root==NULL)
        return 0 ;
     
    int L,R ;
    L=height(root->left) ;
    R=height(root->right) ;
 
    return(max(L,R)+1) ;
}

int max(int a,int b)
{
    return (a>b?a:b) ;
}

---------------------------------------------------------------------------------

Comments

Popular posts from this blog

Count ways to N'th Stair(Order does not matter)

Replace all ‘0’ with ‘5’ in an input Integer

Chocolate Distribution Problem

Remove characters from the first string which are present in the second string

Primality Test ( CodeChef Problem code: PRB01 )