DEV Community

Cover image for 30-Day LeetCoding Challenge: Diameter of Binary Tree
Ahmad Ra'fat
Ahmad Ra'fat

Posted on

30-Day LeetCoding Challenge: Diameter of Binary Tree

The problem


Solution:

It asks us to find a binary tree diameter, it also explains to us that is the diameter of a binary tree is the length of the longest path between any two nodes in a tree.

It is also indicated that doesn't matter if the path passes by the root or not.

So, we need to calculate the depth of each node in the tree and then find the longest path between the nodes.

First the depth of node can be found by node.depth = max(node.left.depth, node.right.depth) + 1.

Also, the path always will be the depth of the left subtree + the depth of the right subtree, so on each node, we need to choose the maximum value that takes us to the final answer so our answer will be: diameter = max(diameter, left_depth + right_depth).

So, how we can calculate that if you look to the illustration above we see that we need to go deep to the tree to get one node depth which means we need to use depth-first search DFS.

We use DFS to find the left subtree depth then find the right subtree depth and later choose between them.

  • Pseudocode:

  • diameterOfBinaryTree(root)

1. define our answer
2. call find_depth_helper(root) with the root to find the longest path.
3. return our answer
Enter fullscreen mode Exit fullscreen mode
  • find_depth_helper(node)
1. if the node is None return 0
2. find_depth_helper(node.left) --> left_depth
3. find_depth_helper(node.right) --> right_depth
4. answer = max(answer, left_depth + right_depth)
5. return the current node depth max(left_depth, right_depth) + 1
Enter fullscreen mode Exit fullscreen mode
  • Complexity:

  • Time Complexity: it is so clear that we go over all our nodes so our time will be O(n) where n is the number of nodes in the tree.

  • Space Complexity: we use recursion stack as we do recursion calls and this stack will be equal to the number of nodes we have so our overall space will be O(n).

  • Solution in python

Top comments (0)