你这个学期必须选修 numCourses
门课程,记为 0
到 numCourses - 1
。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites
给出,其中 prerequisites[i] = [ai, bi]
,表示如果要学习课程 ai
则 必须 先学习课程 bi
。
- 例如,先修课程对
[0, 1]
表示:想要学习课程0
,你需要先完成课程1
。
请你判断是否可能完成所有课程的学习?如果可以,返回 true
;否则,返回 false
。
cpp
//抄的
class Solution {
public:
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
vector<vector<int>> adj(numCourses);
vector<int> inDegree(numCourses,0);
for(auto& edge:prerequisites)
{
adj[edge[1]].push_back(edge[0]);
inDegree[edge[0]]++;
}
queue<int> q;
for(int i=0;i<numCourses;i++)
{
if(inDegree[i]==0)
{
q.push(i);
}
}
int learned = 0;
while(!q.empty())
{
int u = q.front();
q.pop();
learned++;
for(int i:adj[u])
{
if(--inDegree[i]==0)
{
q.push(i);
}
}
}
return learned == numCourses;
}
};
首先构建邻接表和入度数组,入度即指向每个课程的前置课程数,入度为零,表示可以直接学习
逻辑很清晰,首先将所有入度为零的课程入队,出队时,出队的课程指向的课程入度减一,如果变化后入度等于零则入队,这样循环到底,直至队列清空。
循环时记录学习数可以方便最后判断,也可以判断是否还有课程入度为零,多一次循环少一个变量