-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution62.java
36 lines (31 loc) · 1 KB
/
Solution62.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
class Solution {
public List<List<Integer>> getAncestors(int n, int[][] edges) {
List<List<Integer>> res = new ArrayList<>();
for (int i = 0; i < n; i++) {
res.add(new ArrayList<>());
}
ArrayList<Integer>[] graph = new ArrayList[n];
for (int i = 0; i < n; i++) {
graph[i] = new ArrayList<>();
}
for (int[] edge : edges) {
graph[edge[0]].add(edge[1]);
}
for (int i = 0; i < n; i++) {
dfs(graph, i, i, res, new boolean[n]);
}
for (int i = 0; i < n; i++) {
res.get(i).sort(Integer::compareTo);
}
return res;
}
private void dfs(ArrayList<Integer>[] graph, int parent, int curr, List<List<Integer>> res, boolean[] visit) {
visit[curr] = true;
for (int dest : graph[curr]) {
if (!visit[dest]) {
res.get(dest).add(parent);
dfs(graph, parent, dest, res, visit);
}
}
}
}