Diameter Of Binary Tree

Posted: 7 Dec, 2020
Difficulty: Easy

PROBLEM STATEMENT

Try Problem

You are given a Binary Tree. You are supposed to return the length of the diameter of the tree.

The diameter of a binary tree is the length of the longest path between any two end nodes in a tree.

Note :
The number of edges between two nodes represents the length of the path between them.
For Example :
For the given binary tree

Example

Nodes in the diameter are highlighted. The length of the diameter, i.e., the path length, is 6.
Input Format :
The first line contains an integer 'T' which denotes the number of test cases or queries to be run. Then the test cases are as follows.

The first line of each test case contains elements of the tree in the level order form. The line consists of values of nodes separated by a single space. In case a node is null, we take -1 in its place.

tree

For Example, the input for the tree depicted in the above image would be :

1
2 3
4 -1 5 6
-1 7 -1 -1 -1 -1
-1 -1

Explanation :

Level 1 :
The root node of the tree is 1

Level 2 :
Left child of 1 = 2
Right child of 1 = 3

Level 3 :
Left child of 2 = 4
Right child of 2 = null (-1)
Left child of 3 = 5
Right child of 3 = 6

Level 4 :
Left child of 4 = null (-1)
Right child of 4 = 7
Left child of 5 = null (-1)
Right child of 5 = null (-1)
Left child of 6 = null (-1)
Right child of 6 = null (-1)

Level 5 :
Left child of 7 = null (-1)
Right child of 7 = null (-1)
Note :
The above format was just to provide clarity on how the input is formed for a given tree. 
The sequence will be put together in a single line separated by a single space. Hence, for the above-depicted tree, the input will be given as:

1 2 3 4 -1 5 6 -1 7 -1 -1 -1 -1 -1 -1
Output Format :
Print an integer that denotes the length of the diameter of the given binary tree.

Print the output of each test case in a new line.
Note :
You do not need to print anything. It has already been taken care of. Just implement the given function.
Constraints :
1 <= T <= 10
0 <= N <= 10^5
1 <= data <= 10^4

Where ‘N’ is the total number of nodes in the binary tree, and 'data' is the value of the binary tree node.

Time Limit: 1 sec
Approach 1

The basic idea of this approach is to break the problem into subproblems. 

Now, there are three possible cases:

 

  1. The diameter of the tree is present in the left subtree.
  2. The diameter of the tree is present in the right subtree.
  3. The diameter of the tree passes through the root node.

 

Let us define a recursive function, ‘getDiamter’, which takes the root of the binary tree as input parameter and returns the diameter of the given subtree rooted at the “root” node. 

 

We can easily find the diameter of the left subtree and right subtree by recursion. The main task is to calculate the diameter of the tree corresponding to point 3 mentioned above.

 

 

 

From the above figure, we can observe that if the diameter passes through the root node, then it can be written as the length of the longest path between the leaves which passes through the root. And we can get that using the height of the left and right subtrees.

 

Now, assume a function, ‘getHeight,’ which returns the height of the subtree rooted at the “root” node. 

 

The longest path length(i.e., Number of edges in the path) between the leaves can be written as:

1 + getHeight(left child of the root node) + getHeight(right child of the root node)

 

Algorithm 

 

  • If the ‘root’ node is NULL, return 0. The diameter of an empty tree is 0.
  • Recur for the left subtree and store the diameter of the left subtree in a variable ‘leftDiameter’, i.e. ‘leftDiameter’ = getDiameter(left child of the root node)
  • Similarly, recur for the right subtree and store the diameter of the right subtree in a variable ‘rightDiameter’ i.e. ‘rightDiameter’ = getDiameter(right child of the root node)
  • Now, get the height of the left subtree and right subtree and store it in a variable.
    • ‘leftHeight’ = getHeight(left child of the root node)
    • ‘rightHeight’ = getHeight(right child of the root node)
  • The diameter of the given tree will be the maximum of the following terms:
    • ‘leftDiameter’
    • ‘rightDiameter’
    • 1 + ‘leftHeight’ + ‘rightHeight’
  • Return the maximum of above terms i.e.max(leftDiameter, rightDiameter, 1 + leftHeight + rightHeight).
Try Problem