-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path210.java
52 lines (50 loc) · 1.74 KB
/
210.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
import java.util.ArrayList;
import java.util.List;
import java.util.Stack;
class Solution {
class Course {
int id;
List<Integer> prerequisites;
Course(int id) {
this.id = id;
prerequisites = new ArrayList<>();
}
}
public int[] findOrder(int numCourses, int[][] prerequisites) {
List<Course> courses = new ArrayList<>();
for (int i = 0; i < numCourses; i++) {
Course c = new Course(i);
courses.add(c);
}
for (int[] prerequisite : prerequisites) {
Course c = courses.get(prerequisite[0]);
c.prerequisites.add(prerequisite[1]);
}
Stack<Course> stack = new Stack<>();
for (Course course : courses) {
boolean[] visited = new boolean[numCourses];
addToStack(course, stack, courses, visited);
}
if (stack.size() != numCourses) return new int[]{};
int[] res = new int[numCourses];
for (int i = res.length - 1; i >= 0; i--) {
res[i] = stack.pop().id;
}
return res;
}
public boolean addToStack(Course course, Stack<Course> stack, List<Course> courses, boolean[] visited) {
if (visited[course.id]) return false;
visited[course.id] = true;
if (course.prerequisites.isEmpty() && !stack.contains(course)) {
return stack.add(course);
}
for (int prerequisite : course.prerequisites) {
if (stack.contains(courses.get(prerequisite))) continue;
if (!addToStack(courses.get(prerequisite), stack, courses, visited)) {
return false;
}
}
if (!stack.contains(course)) return stack.add(course);
return true;
}
}