Last Updated: 9 Sep, 2021

Maximum Level Sum

Moderate
Asked in company
Google inc

Problem statement

You are given a tree consisting of ‘N’ nodes. Your task is to find the level of the tree with the maximum sum.

The level of a node of the tree is the distance of the node from the root node + 1.

For example:
You are given the tree:

altText

The Sum of all the levels are = [3, 9, 7], Here the level whose sum is maximum is level 2. Hence the answer is 2.
Example:
Elements are in the level order form. The input consists of values of nodes separated by a single space in a single line. In case a node is null, we take -1 in its place.

For example, the input for the tree depicted in the below image would be :

Example

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)

The first not-null node (of the previous level) is treated as the parent of the first two nodes of the current level. 

The second not-null node (of the previous level) is treated as the parent node for the next two nodes of the current level and so on.

The input ends when all nodes at the last level are 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
Input Format:
The first line of the input contains a single integer, 'T,’ denoting the number of test cases.

The first line of each test case contains the elements of the tree in the level order form separated by a single space. If any node does not have a left or right child, take -1 in its place. Refer to the example for further clarification.
Output Format:
For each test case, print a single integer representing the level whose sum is maximum.

Print the output of each test case in a separate line.
Constraints:
1 <= T <= 10
2 <= N <= 10^6
-10^8 <= nodeVal <= 10^8

Time Limit: 1 sec
Note:
You do not need to print anything. It has already been taken care of. Just implement the function.

Approaches

01 Approach

In this approach, we will use depth-first search, and at each call of the dfs we will keep track of the level and add the node value to the level, then we will have an array of level order sum of the whole tree. Then we can find the maximum sum from that level.

 

We create a function dfs(node, levelSum, level), here node is the current node, level is the level of the node, and levelSum is the array to store sums of each level in the tree.
 

Algorithm:

  • The dfs(node,levelSum, level) funtion
    • If node is null
      • Return from the function.
    • If size of levelSum is equal to level
      • Insert node.val into levelSum
    • Otherwise
      • Add node.val to levelSum[level]
    • Call dfs(node.left, levelSum, level + 1)
    • Call dfs(node.right, levelSum, level + 1)
  • Initialise an empty array levelSum
  • Call dfs(root, levelSum, 0)
  • Set maxLevel and maxSum as 0 and -infinity respectively.
  • iterate i through 0 to size of levelSum - 1
    • If levelSum[i] is greater than maxSum
      • Set maxSum as levelSum[i]
      • Set maxLevel as  i + 1
  • Return maxLevel.

02 Approach

In this approach, we will do a level order traversal on the tree and find out the sum of all levels of the tree. Each time we find a sum greater than the current sum, we update the sum and maximum level.

 

Algorithm:

  • Set maxSum as a large minium integer value
  • Set level and maxLevel as 0
  • Initialize an empty queue q
  • Insert root in the q
  • While the q is not empty
    • Increase level by 1
    • Set currSum as 0
    • Iterate i from 0 to size of q
      • Set the front of the q as node and delete the front element from q
      • Add node.val to currSum
      • If node.left is not null
        • Insert node.left in the q
      • If node.right is not null
        • insert node.right in the q
    • If maxSum is greater than currSum
      • Set maxSum as currSum
      • Set maxLevel as level
  • Return maxLevel