题目
你这个学期必须选修 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的课,这些课可以直接学。
- 对于队列中的每一门课,遍历前置课程数组,将前置课程为该课的课,将其前置课程数减一。如果减一后该课的前置课程数为0了,说明可以学了,把他们也加入队列。如果到最后,学完的课程数不等于给定的课程数,那么说明不可能学完。
代码
class Solution {
public:
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
vector<int> indegree( numCourses );
queue<int> q;
// 每门课的入度,即每门课有多少门前置课程
for ( auto& pre : prerequisites )
++indegree[pre[0]];
// 把没有前置课程的课先放入队列
for ( int i = 0; i < indegree.size(); ++i ) {
if ( indegree[i] == 0 )
q.push( i );
}
// 对于以入度为0为前置课程的课程,减少入度。
vector<int> finished; // 保存入度为0的节点,即能学完的课程
while ( !q.empty() ) {
int n = q.front();
q.pop();
finished.push_back( n );
for ( auto& pre : prerequisites ) {
if ( pre[1] == n ) { // pre[0]以pre[1]为前置课程
--indegree[pre[0]];
if ( indegree[pre[0]] == 0 )
q.push( pre[0] );
}
}
}
return finished.size() == numCourses;
}
};