New update is available. Click here to update.

Equal Tree Partition

Posted: 24 Mar, 2021
Difficulty: Moderate


Try Problem

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 :

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