K distance from root
Easy Accuracy: 55.52% Submissions: 51672 Points: 2

Given a Binary Tree of size N and an integer K. Print all nodes that are at distance k from root (root is considered at distance 0 from itself). Nodes should be printed from left to right. If k is more that height of tree, nothing should be printed.

For example, if below is given tree and k is 2. Output should be 4 5 6.

       /     \
     2        3
   /         /   \
  4        5    6 

Example 1:

K = 0
    /   \
   3     2
Output: 1

Example 2:

K = 3
      /  \
     5    3
Output: 5 3

Your Task:
You don't have to take input. Complete the function Kdistance() that accepts root node and as parameter and return the value of the nodes that are at a distance k from the root. (The values are returned as vector in cpp, as ArrayList in java and list in python)

Expected Time Complexity: O(N).
Expected Auxiliary Space: O(Height of the Tree).

1 <= N <= 104

0 <= K <= 30

Note: The Input/Ouput format and Example is given are used for the system's internal purpose, and should be used by a user for Expected Output only. As it is a function problem, hence a user should not read any input from the stdin/console. The task is to complete the function specified, and not to write the full code.

We are replacing the old Disqus forum with the new Discussions section given below.
Click here to view old Disqus comments.

to report an issue on this page.


We strongly recommend solving this problem on your own before viewing its editorial. Do you still want to view the editorial?


All Submissions

My Submissions:

Login to access your submissions.

K distance from root

Output Window