Last Updated: 7 Dec, 2021

Path Sum

Easy

Asked in companies

Solve now

Problem statement

You are given the root node of a binary tree consisting of ‘N’ nodes and an integer value ‘TARGET’. Your task is to find the number of leaf nodes in the given binary tree such that the sum of all nodes from the root to leaf is equal to ‘TARGET’.

A leaf node is defined as a node having no child nodes.

You are given a root node ‘ROOT’.Your task is to return the number of leaf nodes satisfying the given condition.

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 an 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.
The second line of each test case contains an integer corresponding to ‘TARGET’.

Output Format:

For each test case, print the preorder traversal of the given tree.

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 <= N <= 10^6
1 <= nodeVal <=10^9

Time limit: 1 sec

Solve now

Approaches

01 Approach

This approach will create a recursive function REC(‘CUR’,’REQ_SUM’) that will return the number of good nodes in the subtree with root as ‘CUR’ node and ‘REQ_SUM’ is the required sum. We will first check whether ‘CUR’ node is leaf or not and check the required sum then recursively call this function for its left and right subtree to find the number of required leaf nodes in the left and right subtree, respectively.

Algorithm:

Defining the REC(‘CUR’,’ REQ_SUM’)
If the ‘’CUR’ is an empty node, return 0.
If CUR’s left is empty and CUR’s right is empty:
- CUR is a leaf node.
- If REQ_SUM is equal to CUR’s value:
  - Return 1.
- Else:
  - Return 0.
Set ‘ANS’ as 0.
Now, update ‘REQ_SUM’ as ‘REQ_SUM’ - CUR’s value.
Set ‘ANS’ as ‘ANS’ + REC(left node of ‘CUR’, REQ_SUM).
Set ‘ANS’ as ‘ANS’ + REC(right node of ‘CUR’, REQ_SUM).
Return ‘ANS’.

Set ‘ANS’ as REC(‘ROOT’, ‘TARGET’).
Return ‘ANS’.

Solve now

02 Approach

In this approach, we will traverse each node iteratively and check whether the node is good or not. We will use a stack ‘NODE_STACK’ to maintain the order of traversal. We will insert the ‘ROOT’ node into the ‘NODE_STACK’.

We will iterate till ‘NODE_STACK’ is not empty -:

We will delete the top node from the stack, check whether the node is a leaf with the required sum as 0.
Update the ‘REQ_SUM’.
We will insert its left and right child into NODE_STACK to traverse the left and right subtree.

At last, we will return the ANS as the number of good nodes.

Algorithm:

We will declare a stack 'NODE_STACK' to maintain the order of traversal.
Declare a variable ‘ANS’ to store the number of good nodes.
Insert [‘ROOT’,’TARGET’] into 'NODE_STACK'.
While 'NODE_STACK' is not empty, do the following steps:
- Set cur as a top node of the 'NODE_STACK'.
- Delete cur from 'NODE_STACK'.
- Set ‘CUR_NODE’ as CUR[0].
- Set ‘REQ_SUM’ as CUR[1].
- If CUR_NODE is leaf and ‘REQ_SUM’ is equal to CUR_NODE’s val:
  - Required leaf node found.
  - Set ‘ANS’ as ‘ANS’+1.
- Now, update ‘REQ_SUM’ as ‘REQ_SUM’ - CUR’s value.
- If the left node of CUR_NODE is not empty:
  - Push [left of CUR_NODE,MX] into ‘NODE_STACK’.
- If the right node of CUR_NODE is not empty:
  - Push [right of CUR_NODE,MX] into ‘NODE_STACK’.
Return ‘ANS’.

Solve now