-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathCourse Schedule II.cpp
61 lines (52 loc) · 2.45 KB
/
Course Schedule II.cpp
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
/* Leet Code */
/* Title - Course Schedule II */
/* Created By - Akash Modak */
/* Date - 18/7/2020 */
// 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.
// Example 1:
// Input: 2, [[1,0]]
// Output: [0,1]
// Explanation: 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] .
// Example 2:
// Input: 4, [[1,0],[2,0],[3,1],[3,2]]
// Output: [0,1,2,3] or [0,2,1,3]
// Explanation: 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.
// You may assume that there are no duplicate edges in the input prerequisites.
class Solution {
public:
bool cycle_detect(int n, vector<int> &topological_order, vector<vector<int> > &adj) {
vector<int> degree(n, 0);
for (vector<int> p: adj)
for(int node: p)
degree[node]++;
queue<int> q;
for (int i = 0; i < n; i++)
if (degree[i] == 0) q.push(i);
while (!q.empty())
{
int curr = q.front(); q.pop(); n--;
topological_order.push_back(curr);
for (auto next: adj[curr])
if (--degree[next] == 0) q.push(next);
}
return n == 0;
}
vector<int> findOrder(int numCourses, vector<vector<int> >& prerequisites) {
vector<int> topologicalOrder;
vector<vector<int> > adj(numCourses);
for(vector<int> v: prerequisites){
adj[v[1]].push_back(v[0]);
}
if(!cycle_detect(numCourses, topologicalOrder, adj))
return vector<int>(0);
return topologicalOrder;
}
};