210. 课程表 II/C++

Lambda_eta

于 2019-07-05 16:10:46 发布

阅读量522

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode/C++

本文链接：https://blog.youkuaiyun.com/Zolewit/article/details/94739077

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文深入探讨了一种用于解决课程先修顺序问题的图算法。通过使用深度优先搜索(DFS)来检测图中是否存在环，从而判断课程之间的依赖关系是否构成有效的学习路径。如果从每个节点开始的DFS都能成功完成，则表明所有课程都可以被正确地排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

class Solution {
public:
    vector<int> res;
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        if(numCourses==0) return {};
        
        vector<vector<int>> tmp(numCourses);
        //flag：2表示从该节点起无环，1表示正在遍历，0表示还未遍历过
        vector<int> flag(numCourses);
        
        //tmp[i]里包含所有i指向的节点
        for(int i=0;i<prerequisites.size();++i){
            tmp[prerequisites[i][0]].push_back(prerequisites[i][1]);
        }
        
        bool ans = true;
        for(int i=0;i<numCourses;++i){
            ans = ans&&dfs(i,flag,tmp);
        }
        if(ans){
            for(int i=0;i<flag.size();++i){
                if(!flag[i]){
                    res.push_back(i);
                }
            }
            return res;
        }
        return {};
    }
    
    bool dfs(int i, vector<int> &flag, const vector<vector<int>> &tmp){
        if(flag[i]==2)//之前已检验过，从该点起无环，直接返回
            return true;
        else if(flag[i]==1)//是正在检验的点，说明有环
            return false;
        
        flag[i] = 1;//表示正在检验
        for(int j=0;j<tmp[i].size();++j){
            if(dfs(tmp[i][j],flag,tmp)){
                continue;
            }
            return false;
        }
        res.push_back(i);
        flag[i] = 2;
        return true;
    }
};