K_distance_from_root.java

// { Driver Code Starts
//Initial Template for Java

import java.util.LinkedList;
import java.util.Queue;
import java.io.*;
import java.util.*;

class Node {
  int data;
  Node left;
  Node right;

  Node(int data) {
    this.data = data;
    left = null;
    right = null;
  }
}

class GfG {

  static Node buildTree(String str) {

    if (str.length() == 0 || str.charAt(0) == 'N') {
      return null;
    }

    String ip[] = str.split(" ");
    // Create the root of the tree
    Node root = new Node(Integer.parseInt(ip[0]));
    // Push the root to the queue

    Queue<Node> queue = new LinkedList<>();

    queue.add(root);
    // Starting from the second element

    int i = 1;
    while (queue.size() > 0 && i < ip.length) {

      // Get and remove the front of the queue
      Node currNode = queue.peek();
      queue.remove();

      // Get the current node's value from the string
      String currVal = ip[i];

      // If the left child is not null
      if (!currVal.equals("N")) {

        // Create the left child for the current node
        currNode.left = new Node(Integer.parseInt(currVal));
        // Push it to the queue
        queue.add(currNode.left);
      }

      // For the right child
      i++;
      if (i >= ip.length)
        break;

      currVal = ip[i];

      // If the right child is not null
      if (!currVal.equals("N")) {

        // Create the right child for the current node
        currNode.right = new Node(Integer.parseInt(currVal));

        // Push it to the queue
        queue.add(currNode.right);
      }
      i++;
    }

    return root;
  }

  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

    int t = Integer.parseInt(br.readLine());

    while (t > 0) {
      String X[] = br.readLine().trim().split(" ");
      int k = Integer.parseInt(X[0]);
      String s = br.readLine();
      Node root = buildTree(s);
      Tree g = new Tree();
      ArrayList<Integer> nodes = g.Kdistance(root, k);
      for (int i = 0; i < nodes.size(); i++) {
        System.out.print(nodes.get(i) + " ");
      }
      System.out.println();
      t--;

    }
  }

}

class Tree {
  ArrayList<Integer> Kdistance(Node root, int k) {
    ArrayList<Integer> ans = new ArrayList<>();
    KdistanceUtil(root, k, ans);
    return ans;
  }

  void KdistanceUtil(Node root, int k, ArrayList<Integer> ans) {
    if (root == null) {
      return;
    }
    if (k == 0) {
      ans.add(root.data);
      return;
    } else {
      KdistanceUtil(root.left, k - 1, ans);
      KdistanceUtil(root.right, k - 1, ans);
    }
  }
}