-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathSolution.java
67 lines (48 loc) · 2.28 KB
/
Solution.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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.Set;
import java.util.Map.Entry;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import java.util.HashMap;
import java.util.LinkedList;
public class Solution {
private static String solve(Map<String, List<String>> prerequisitesMap) {
List<String> answer = new ArrayList<>(prerequisitesMap.entrySet().size());
Map<String, List<String>> prerequisiteOfMap = new HashMap<>();
Queue<String> coursesWithoutPrerequesite = new LinkedList<>();
for (String course : prerequisitesMap.keySet()) {
List<String> prerequisites = prerequisitesMap.get(course);
if (prerequisites.size() == 0)
coursesWithoutPrerequesite.add(course);
for (String prerequisite : prerequisites) {
prerequisiteOfMap.computeIfAbsent(prerequisite, c -> new ArrayList<>()).add(course);
}
}
while (!coursesWithoutPrerequesite.isEmpty()) {
String courseWithoutPrerequisite = coursesWithoutPrerequesite.poll();
answer.add(courseWithoutPrerequisite);
if (!prerequisiteOfMap.containsKey(courseWithoutPrerequisite))
continue;
for (String course : prerequisiteOfMap.get(courseWithoutPrerequisite)) {
List<String> filteredPrerequisites = prerequisitesMap.get(course).stream()
.filter(p -> p != courseWithoutPrerequisite).collect(Collectors.toList());
if (filteredPrerequisites.size() == 0)
coursesWithoutPrerequesite.add(course);
prerequisitesMap.put(course, filteredPrerequisites);
}
prerequisiteOfMap.remove(courseWithoutPrerequisite);
}
return answer.toString();
}
public static void main(String[] args) {
Map<String, List<String>> prerequisitesMap = new HashMap<>();
prerequisitesMap.put("CSC300", Arrays.asList("CSC100", "CSC200"));
prerequisitesMap.put("CSC200", Arrays.asList("CSC100"));
prerequisitesMap.put("CSC100", Arrays.<String>asList());
System.out.println(solve(prerequisitesMap)); // [CSC100, CSC200, CSC300]
}
}