Amazon Interview Question for SDE-2s

Country: United States

Comment hidden because of low score. Click to expand.
of 1 vote

Level order bfs

- Ashu July 22, 2020 | Flag Reply
Comment hidden because of low score. Click to expand.
of 0 vote

use bfs/level order traversal while keeping track of a distance from a root. If the first occurence of each number of the given array is found, update the result array with the current distance.
time O(n) where n is the total number of nodes in a tree
space O(Math.max(k, n)) where k is the total number of elements in an array.

public static int[] getDistanceFromRoot(Node head, int[] arr) {
 int[] result = new int[arr.length];
 for (int i = 0; i < result.length; ++i) {
  result[i] = -1;

 HashMap<Integer, Integer> hm = new HashMap<>();
 for (int j = 0; j < arr.length; ++j) {
  hm.put(arr[j], j);

 Queue<Node> q = new LinkedList<>();
 int level = 0;

 while (!q.isEmpty()) {
  int size = q.size();
  while (size > 0) {
   Node temp = q.remove();
   if (hm.containsKey(temp.val) && result[hm.getValue(temp.val)] == -1) {
    result[hm.get(temp.val)] = level;


 return result;

- Euihoon Seol October 18, 2020 | Flag Reply

Add a Comment

Writing Code? Surround your code with {{{ and }}} to preserve whitespace.


is a comprehensive book on getting a job at a top tech company, while focuses on dev interviews and does this for PMs.

Learn More


CareerCup's interview videos give you a real-life look at technical interviews. In these unscripted videos, watch how other candidates handle tough questions and how the interviewer thinks about their performance.

Learn More

Resume Review

Most engineers make critical mistakes on their resumes -- we can fix your resume with our custom resume review service. And, we use fellow engineers as our resume reviewers, so you can be sure that we "get" what you're saying.

Learn More

Mock Interviews

Our Mock Interviews will be conducted "in character" just like a real interview, and can focus on whatever topics you want. All our interviewers have worked for Microsoft, Google or Amazon, you know you'll get a true-to-life experience.

Learn More