Challenge: Find Nodes at "k" distance from the Root

If you are given the root to a Binary Search Tree and a value "k", can you write a code to find the nodes at "k" distance from the root? A solution is placed in the "solution" section for your help, but we would suggest you to solve it on your own first.

We'll cover the following

Problem Statement

Implement a function findKNodes(root,k) which finds and returns nodes at k distance from the root in the given binary tree. An illustration is also provided for your understanding.

Output

Returns all nodes in a list format which are at k distance from the root node.

Sample Input

bst = {
	6 -> 4,9
    4 -> 2,5
    9 -> 8,12
    12 -> 10,14
}
where parent -> leftChild,rightChild
k = 2

Sample Output

[2,5,8,12]

Level up your interview prep. Join Educative to access 70+ hands-on prep courses.