Last Updated: 6 Dec, 2021

Good Nodes

Easy

Asked in companies

Solve now

Problem statement

You are given the root node of a binary tree consisting of ‘N’ nodes. Your task is to find the number of good nodes in the given binary tree.

A good node is defined as good if there are no nodes with a value greater than X’s value in the path from the root to X.

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

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.

Output Format:

For each test case, print an integer representing the total number of good nodes.

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^5

Time limit: 1 sec

Solve now

Approaches

01 Approach

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

Algorithm:

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

Set ‘ANS’ as REC(‘ROOT’, the value of ‘ROOT’).
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 good or not.
Update the ‘MX’.
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’,ROOT’s value] 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 ‘MX’ as CUR[1].
- If CUR_NODE’ s value is greater or equal to ‘MX’:
  - Good Node found.
  - Set ‘ANS’ as ‘ANS’+1.
- Update ‘MX’ to maximum of ‘MX’ and ‘CUR_NODE ’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