-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPOTD-Nodes at given distance in binary tree.java
48 lines (40 loc) · 1.27 KB
/
POTD-Nodes at given distance in binary tree.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
class Solution
{
public static ArrayList<Integer> KDistanceNodes(Node root, int target , int k)
{
// return the sorted list of all nodes at k dist
ArrayList<Integer> list = new ArrayList<>();
findKDN(root,target,k,list);
Collections.sort(list);
return list;
}
static void findKDNdown(Node node, int k, ArrayList<Integer> list){
if(node==null || k<0) return;
if(k==0){
list.add(node.data);
return;
}
findKDNdown(node.left,k-1,list);
findKDNdown(node.right,k-1,list);
}
static int findKDN(Node node, int target, int k, ArrayList<Integer> list){
if(node==null) return -1;
if(node.data==target){
findKDNdown(node,k,list);
return 0;
}
int ld = findKDN(node.left,target,k,list);
if(ld!=-1){
if(ld+1==k) list.add(node.data);
else findKDNdown(node.right,k-ld-2,list);
return ld+1;
}
int rd = findKDN(node.right,target,k,list);
if(rd!=-1){
if(rd+1==k) list.add(node.data);
else findKDNdown(node.left,k-rd-2,list);
return rd+1;
}
return -1;
}
};