210. Course Schedule II(课程表 II)
题目大意
There are a total of numCourses
courses you have to take, labeled from 0 to numCourses - 1
. You are given an array prerequisites
where prerequisites[i] = [ai, bi]
indicates that you must take course bi
first if you want to take course ai
.
For example, the pair [0, 1]
, indicates that to take course 0 you have to first take course 1.
Return the ordering of courses you should take to finish all courses. If there are many valid answers, return any of them. If it is impossible to finish all courses, return an empty array.
中文释义
你总共要上 numCourses
门课程,从 0 到 numCourses - 1
进行标记。你会得到一个数组 prerequisites
,其中 prerequisites[i] = [ai, bi]
表示如果你想上 ai
课程,你必须先上 bi
课程。
例如,对 [0, 1]
这对来说,意味着要上 0 课程,你必须先上 1 课程。
返回你应该上的课程顺序以完成所有课程。如果有多个有效答案,返回其中任何一个。如果不可能完成所有课程,返回一个空数组。
示例
- 示例 1:
- 输入:
numCourses = 2
,prerequisites = [[1,0]]
- 输出:
[0,1]
- 解释:总共有 2 门课程要上。要上课程 1,你应该已经完成了课程 0。所以正确的课程顺序是 [0,1]。
- 输入:
- 示例 2:
- 输入:
numCourses = 4
,prerequisites = [[1,0],[2,0],[3,1],[3,2]]
- 输出:
[0,2,1,3]
- 解释:总共有 4 门课程要上。要上课程 3,你应该已经完成了课程 1 和 2。课程 1 和 2 应该在你完成课程 0 之后上。
所以一个正确的课程顺序是 [0,1,2,3]。另一个正确的排序是 [0,2,1,3]。
- 输入:
- 示例 3:
- 输入:
numCourses = 1
,prerequisites = []
- 输出:
[0]
- 输入:
限制条件
1 <= numCourses <= 2000
0 <= prerequisites.length <= numCourses * (numCourses - 1)
prerequisites[i].length == 2
0 <= ai, bi < numCourses
ai != bi
- 所有
[ai, bi]
对都是不同的。
解题思路
使用拓扑排序来解决问题。首先建立图和入度表,然后使用队列进行拓扑排序,得出完成所有课程的顺序。
步骤说明
- 构建一个图,其中每个节点表示一个课程,边表示先修关系。
- 初始化一个入度表,记录每个课程的先修课数量。
- 将所有入度为 0 的课程加入队列。
- 进行拓扑排序:
- 每次从队列中取出一个课程,将其加入结果列表。
- 递减该课程所有邻接课程的入度。如果邻接课程入度为 0,则将其加入队列。
- 如果结果列表的大小不等于课程总数,返回空数组;否则返回结果列表。
代码
class Solution {
public:
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vector<vector<int>> graph(numCourses);
vector<int> inDegree(numCourses, 0);
vector<int> order;
queue<int> myqueue;
for (int i = 0; i < prerequisites.size(); i++) {
graph[prerequisites[i][1]].push_back(prerequisites[i][0]);
inDegree[prerequisites[i][0]]++;
}
for (int i = 0; i < numCourses; i++) {
if (inDegree[i] == 0) myqueue.push(i);
}
while (!myqueue.empty()) {
int top = myqueue.front();
myqueue.pop();
order.push_back(top);
for (int i = 0; i < graph[top].size(); i++) {
inDegree[graph[top][i]]--;
if (inDegree[graph[top][i]] == 0) myqueue.push(graph[top][i]);
}
}
if (order.size() != numCourses) {
return {};
} else {
return order;
}
}
};