图:207课程表 题解:入度数组,邻接表,队列,拓扑排序
没做出来,参考题解,这篇题解写的非常好。
把一个 有向无环图 转成 线性的排序 就叫 拓扑排序。(没太懂这句话的意思)
class Solution {
public:
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
vector<int> inDegree(numCourses, 0);//入度数组
unordered_map<int, vector<int>> map; //邻接表
//初始化入度数组和邻接表
for(const auto& pre : prerequisites){
inDegree[pre[0]]++; //计算每个课程的入度
map[pre[1]].push_back(pre[0]); //建立邻接表
}
queue<int> q;
//将所有入度为0的课程加入队列
for(int i = 0; i < numCourses; i++){
if(inDegree[0] == 0){
q.push(i);
}
}
int count = 0;//选课计数
//开始拓扑排序
while(!q.empty()){
int selected = q.front(); //选出当前的课
q.pop();
count++;//选课数+1
//获取依赖这门课的后续课程
if(map.find(selected) != map.end()){
for(int nextCourse : map[selected]){
inDegree[nextCourse]--; //减少后续课程的入度
if(inDegree[nextCourse] == 0){
q.push(nextCourse); //如果入度为0,则加入队列
}
}
}
}
return count == numCourses; //如果选完的课数等于总课数,返回true
}
};
浙公网安备 33010602011771号