leetcode原题链接:课程表
上一篇:HOT52-腐烂的橘子
题目描述
你这个学期必须选修 numCourses
门课程,记为 0
到 numCourses - 1
。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites
给出,其中 prerequisites[i] = [ai, bi]
,表示如果要学习课程 ai
则 必须 先学习课程 bi
。
- 例如,先修课程对
[0, 1]
表示:想要学习课程0
,你需要先完成课程1
。
请你判断是否可能完成所有课程的学习?如果可以,返回 true
;否则,返回 false
。
示例 1:
输入:numCourses = 2, prerequisites = [[1,0]] 输出:true 解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。
示例 2:
输入:numCourses = 2, prerequisites = [[1,0],[0,1]] 输出:false 解释:总共有 2 门课程。学习课程 1 之前,你需要先完成课程 0 ;并且学习课程 0 之前,你还应先完成课程 1 。这是不可能的。
提示:
1 <= numCourses <= 105
0 <= prerequisites.length <= 5000
prerequisites[i].length == 2
0 <= ai, bi < numCourses
prerequisites[i]
中的所有课程对 互不相同
解题方法:采用拓扑排序法。
第1步:建立一个邻接表和入度表。v0->v1,v0->v2,则neibor[v0]=[v1, v2]。
第2步:将入度为0的节点(如v0)压入队列中,然后更新这些入度为0的节点的邻接节点入度。更新后如果存在入度为0的邻接节点(如v1, v2),则将其压入队列中。
第3步:用一个结果数组result保存所有入度为0的节点。
第4步: 判断result的个数是否等于节点的个数,等于在返回true,否则返回false。
C++代码
#include <iostream>
#include <vector>
#include <queue>
/*
* v0 <-- v1
*
*
*/
class Solution {
public:
bool canFinish(int numCourses, std::vector<std::vector<int>>& prerequisites) {
// 先根据依赖关系,创建邻接图
std::vector<std::vector<int>> neibors(numCourses, std::vector<int>());//保存以v1为入度的邻居节点
std::vector<int> in_degree(numCourses, 0);//保存每一个节点的入度,初始值为0
std::vector<int> result;//保存输出的入度为0的节点
for (const auto& vp : prerequisites) {
auto v0 = vp[0];
auto v1 = vp[1];
neibors[v1].emplace_back(v0);
in_degree[v0]++;
}
std::queue<int> q;
// 开始拓扑排序, 将入度为0的节点v0放入队列中
for (int i = 0; i < numCourses; i++) {
if (in_degree[i] == 0) {
q.emplace(i);
result.emplace_back(i);
}
}
// 更新以v0为初度的邻居节点的入度减1,并再次将入度为0的节点放入队列中
while (!q.empty()) {
int v1 = q.front();
q.pop();
for (auto v0 : neibors[v1]) {
in_degree[v0]--;
if (in_degree[v0] == 0) {
q.emplace(v0);
result.emplace_back(v0);
}
}
}
// 判断最后队列是否为空
return result.size() == numCourses;
}
};