Last Updated: 24 Mar, 2021

Equal Tree Partition

Moderate

Asked in company

Solve now

Problem statement

Ninja has been given a binary tree ‘BINARY_TREE’ of type integer with ‘N’ nodes. Ninja has to check if it is possible to partition the binary tree into two trees with equal sum values after removing exactly one edge from the given binary tree.

As Ninja is busy, he asks you for help. Can you help him determine if an equal sum partition of the given ‘BINARY_TREE’ is possible? Return ‘True’ if possible else return ‘False’.

For example:

In this given ‘BINARY_TREE’, we can make two possible partitions:
1. The first one is with sums 2 and 4.
2. The second one is with sums 3 and 3.

The second partition divides the given tree into an equal sum partition tree, so we return “True”.

Input Format

The first line of input contains an integer 'T' representing the number of test cases. Then the test cases follow.

The next line of each test case contains elements 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.

The next line of each test case contains one integer ‘K’ representing that the houses that are still not vaccinated are at a ‘K’ distance from the last houses in the city.

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

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

Explanation :

Level 1 :
The root node of the tree is 4

Level 2 :
Left child of 4 = 2
Right child of 4 = 6

Level 3 :
Left child of 2 = 1
Right child of 2 = 3
Left child of 6 = 5
Right child of 6 = 7

Level 4 :
Left child of 1 = null (-1)
Right child of 1 = null (-1)
Left child of 3 = null (-1)
Right child of 3 = null (-1)
Left child of 5 = null (-1)
Right child of 5 = null (-1)
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:

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

Output Format :

For each test case, return “True” if we can partition the given ‘BINARY_TREE’ into two trees with equal sum values else return “False”.

Print the output of each test case in a separate 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
1 <= ‘DATA’ <= 10^5 

Time Limit: 1 second

Solve now

Approaches

01 Approach

As we know we can remove exactly on edge from the given binary tree. So first, we calculate the total sum of all nodes of the ‘BINARY_TREE’. If the sum is odd, that means we cannot divide the ‘BINARY_TREE’ into two trees with equal sum values, so we return “false”. Else we traverse the tree and at every node, we check if the sum of the subtree is equal to the total sum of all nodes by 2. If it is true, then we return true else we repeat this for another subtree.

Here is the algorithm:

First, we traverse the tree and find the total sum of all nodes of the ‘BINARY_TREE’ and store it into ‘TOTAL_SUM’.
If ‘TOTAL_SUM’ is odd:
1. Return false.
We traverse the whole ‘BINARY_TREE’ and at every node, we calculate the sum of nodes of the left subtree and the right subtree.
‘CURR_SUM’ = sumOfSubTreeHelper(root).
If the ‘CURR_SUM’ is equal to ‘TOTAL_SUM’ by 2:
1. Return true.
Finally, we return false.

sumOfSubTreeHelper(‘ROOT’)

If ‘ROOT’ is equal to ‘NULL’:
1. Return 0.
‘TOTAL_SUM’ = ‘ROOT’.data + sumOfSubTreeHelper(‘ROOT’.left) + sumOfSubTreeHelper(‘ROOT’.right).
Finally, return ‘TOTAL_SUM’.

Solve now

02 Approach

First, we calculate the total sum of every node of the ‘BINARY_TREE’. Then we traverse the ‘BINARY_TREE’ again and at every node, except the root node, we check if the current sum of the current subtree is equal to the total sum of the ‘BINARY_TREE’ by 2.

Here is the algorithm:

We declare a global variable ‘ANSWER’ in which we store ‘true’ if we partition the ‘BINARY_TREE’ into two trees with equal sum values.
First, we traverse the tree and find the total sum of all nodes of the ‘BINARY_TREE’ and store it into ‘TOTAL_SUM’.
If ‘TOTAL_SUM’ is odd:
1. Return false.
We declare a variable ‘CURR_SUM’ in which we store the sum of the current subtree.
‘CURR_SUM’ = ‘ROOT’.data + checkEqual(root.left) + checkEqual(root.right).
If ‘CURR_SUM’ is equal to ‘TOTAL_SUM’ by 2:
1. ‘ANSWER’ = ‘true’
2. Return 0.
Return ‘CURR_SUM’.

Solve now