-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCourseScheduleII.cpp
More file actions
34 lines (34 loc) · 1.01 KB
/
CourseScheduleII.cpp
File metadata and controls
34 lines (34 loc) · 1.01 KB
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
class Solution {
public:
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vector<int> in_degree(numCourses), ans;
vector<vector<int>> adj(numCourses);
for (vector<int>& prerequisite : prerequisites) {
adj[prerequisite[1]].push_back(prerequisite[0]);
++in_degree[prerequisite[0]];
}
ans.reserve(numCourses);
queue<int> q;
for (int i = 0; i < numCourses; ++i) {
if (!in_degree[i]) {
q.push(i);
ans.push_back(i);
}
}
while (!q.empty()) {
int curr = q.front();
q.pop();
for (int next : adj[curr]) {
// No more prerequisites for this class.
if (!--in_degree[next]) {
q.push(next);
ans.push_back(next);
}
}
}
if (ans.size() != numCourses) {
ans.clear();
}
return ans;
}
};