题目大意
你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-1 。
在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们:[0,1]
给定课程总量以及它们的先决条件,请你判断是否可能完成所有课程的学习?
示例 1:
输入: 2, [[1,0]]
输出: true
解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。
示例 2:
输入: 2, [[1,0],[0,1]]
输出: false
解释: 总共有 2 门课程。学习课程 1 之前,你需要先完成课程 0;并且学习课程 0 之前,你还应先完成课程 1。这是不可能的。
解题思路
有向图问题。首先记录一下每个课程的关联课程,关联课程指的是必须学完当前课程后才可能继续学习的课程,然后再记录一下每个课程的度,即学完几个课程后才能学习当前课程。
创建一个队列,找到度为0的课程(即该课程可直接学习,不依赖其它课程)放入队列。
不断从队列中弹出当前课程,遍历所有依赖当前课程的课程,将对应的度-1,如果度为0,表示不再依赖其它课程,放入队列中。
最后判断学完的课程是否等于总课程。
class Solution {
public:
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
vector<vector<int>> neibors(numCourses);
vector<int> degree(numCourses, 0);
// 记录课程之间的依赖关系
for (vector<int> iter : prerequisites){
++degree[iter[0]];
neibors[iter[1]].push_back(iter[0]);
}
queue<int> courseId;
// 找到初始情况下不依赖其它课程的课程
for (int i = 0; i < numCourses; ++i){
if (degree[i] == 0)
courseId.push(i);
}
// 当前学完的课程
int finishedCourse = 0;
while (!courseId.empty()){
int id = courseId.front(); courseId.pop();
++finishedCourse;
for (int i = 0; i < neibors[id].size(); ++i){
--degree[neibors[id][i]];
if (degree[neibors[id][i]] == 0){
courseId.push(neibors[id][i]);
}
}
}
return finishedCourse == numCourses;
}
};