图:207课程表 题解:入度数组,邻接表,队列,拓扑排序

207. 课程表 - 力扣(LeetCode)

没做出来,参考题解,这篇题解写的非常好。

把一个 有向无环图 转成 线性的排序 就叫 拓扑排序。(没太懂这句话的意思)

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
    }
};

 

posted @ 2024-09-15 20:42  axuu  阅读(0)  评论(0编辑  收藏  举报