mirror of
https://github.com/ShusenTang/LeetCode.git
synced 2024-09-02 14:20:01 +00:00
72 lines
2.3 KiB
Markdown
72 lines
2.3 KiB
Markdown
![]() |
# [210. Course Schedule II](https://leetcode.com/problems/course-schedule-ii/)
|
|||
|
# 思路
|
|||
|
这题是`207. Course Schedule`的变体,207题要求判断是否可以完成整个课程,即判断有向图是否有环,然后我们用拓扑排序来判断是否有环。
|
|||
|
而这题其实就是求有向图的拓扑排序,也分为BFS和DFS两种,思路和207一样的,所以这里不再赘述,可参考[207题解](https://github.com/ShusenTang/LeetCode/blob/master/solutions/207.%20Course%20Schedule.md)。
|
|||
|
|
|||
|
# C++
|
|||
|
## BFS
|
|||
|
``` C++
|
|||
|
class Solution {
|
|||
|
public:
|
|||
|
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
|
|||
|
int arc_num = prerequisites.size();
|
|||
|
stack<int>stk;
|
|||
|
vector<int>in_degree(numCourses, 0);
|
|||
|
vector<vector<int>>G(numCourses, vector<int>{});
|
|||
|
vector<int>res;
|
|||
|
|
|||
|
for(auto &arc : prerequisites){ // 建图
|
|||
|
in_degree[arc[0]]++;
|
|||
|
G[arc[1]].push_back(arc[0]);
|
|||
|
}
|
|||
|
|
|||
|
for(int i = 0; i < numCourses; i++) // 先将所有入度为0的顶点入栈
|
|||
|
if(!in_degree[i]) stk.push(i);
|
|||
|
|
|||
|
while(!stk.empty()){
|
|||
|
int course = stk.top(); stk.pop();
|
|||
|
res.push_back(course);
|
|||
|
for(int c: G[course]){ // 所有以course为起点的边的终点
|
|||
|
if(!(--in_degree[c])) stk.push(c);
|
|||
|
}
|
|||
|
}
|
|||
|
return res.size() == numCourses ? res : vector<int>{};
|
|||
|
}
|
|||
|
};
|
|||
|
```
|
|||
|
|
|||
|
## DFS
|
|||
|
``` C++
|
|||
|
class Solution {
|
|||
|
private:
|
|||
|
bool DFS(vector<vector<int>>&G, vector<int>& visited, vector<int>& res, int i) {
|
|||
|
if (visited[i] == -1) return false;
|
|||
|
if (visited[i] == 1) return true;
|
|||
|
visited[i] = -1;
|
|||
|
for (auto a : G[i]) {
|
|||
|
if (!DFS(G, visited, res, a)) return false;
|
|||
|
}
|
|||
|
res.push_back(i);
|
|||
|
visited[i] = 1;
|
|||
|
return true;
|
|||
|
}
|
|||
|
public:
|
|||
|
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
|
|||
|
vector<vector<int>> G(numCourses, vector<int>());
|
|||
|
vector<int>visited(numCourses, 0);
|
|||
|
vector<int>res;
|
|||
|
|
|||
|
for (auto &arc : prerequisites) {
|
|||
|
G[arc[1]].push_back(arc[0]);
|
|||
|
}
|
|||
|
|
|||
|
for (int i = 0; i < numCourses; ++i)
|
|||
|
if (!DFS(G, visited, res, i)) return vector<int>{};
|
|||
|
|
|||
|
// 注意最后要翻转一下
|
|||
|
reverse(res.begin(), res.end());
|
|||
|
return res;
|
|||
|
}
|
|||
|
};
|
|||
|
```
|