-
Notifications
You must be signed in to change notification settings - Fork 0
/
210_CourseScheduleII.java
58 lines (47 loc) · 2.25 KB
/
210_CourseScheduleII.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
// There are a total of n courses you have to take, labeled from 0 to n - 1.
// Some courses may have prerequisites, for example to take course 0 you have to first take course 1, which is expressed as a pair: [0,1]
// Given the total number of courses and a list of prerequisite pairs, return the ordering of courses you should take to finish all courses.
// There may be multiple correct orders, you just need to return one of them. If it is impossible to finish all courses, return an empty array.
// For example:
// 2, [[1,0]]
// There are a total of 2 courses to take. To take course 1 you should have finished course 0. So the correct course order is [0,1]
// 4, [[1,0],[2,0],[3,1],[3,2]]
// There are a total of 4 courses to take. To take course 3 you should have finished both courses 1 and 2. Both courses 1 and 2 should be taken after you finished course 0. So one correct course order is [0,1,2,3]. Another correct ordering is[0,2,1,3].
// Note:
// The input prerequisites is a graph represented by a list of edges, not adjacency matrices. Read more about how a graph is represented.
//bfs
public class Solution {
public int[] findOrder(int numCourses, int[][] prerequisites) {
int[] res = new int[numCourses];
int index = 0;
int cnt = numCourses;
Map<Integer, List<Integer>> map = new HashMap<Integer, List<Integer>>();
Queue<Integer> queue = new LinkedList<>();
int[] indegree = new int[numCourses];
for(int i=0; i<numCourses; i++) {
map.put(i, new ArrayList<Integer>());
}
for(int i=0; i<prerequisites.length; i++) {
map.get(prerequisites[i][1]).add(prerequisites[i][0]);
indegree[prerequisites[i][0]]++;
}
for(int i=0; i<numCourses; i++) {
if(indegree[i] == 0) {
res[index++] = i;
queue.offer(i);
}
}
while(!queue.isEmpty()) {
int current = queue.poll();
for(int i : map.get(current)) {
if(--indegree[i] == 0) {
queue.offer(i);
res[index++] = i;
}
}
cnt--;
}
if(cnt != 0) return new int[0];;
return res;
}
}