Leetcode---207. 课程表---每日一题---拓扑排序(DFS)

207. 课程表

你这个学期必须选修 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。这是不可能的。

提示:

  1. 输入的先决条件是由 边缘列表 表示的图形,而不是 邻接矩阵 。详情请参见图的表示法。
  2. 你可以假定输入的先决条件中没有重复的边。
  3. 1 <= numCourses <= 10^5

拓扑排序(DFS)

class Solution {
public:
    vector<vector<int>> edge;
    vector<int> vis;
    bool ans = true;
    
    void dfs(int now){
        vis[now] = 1;
        for(int to : edge[now]){
            // 若这个点的状态未变成2(未锁上,即存在环)
            if(vis[to] == 1) {
                ans = false;
                return;
            }
            else if(!vis[to]) {
                dfs(to);
                if(!ans) return;
            }
        }
        vis[now] = 2;
    }

    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        edge.resize(numCourses);
        vis.resize(numCourses);
        for(auto e : prerequisites) {
            edge[e[1]].push_back(e[0]);
        }
        for(int i = 0; i < numCourses; i++) {
            if(!vis[i]) dfs(i);
            if(!ans) break;
        }
        return ans;
    }
};

你可能感兴趣的:(Leetcode,每日一练,拓扑排序,dfs,Leetcode,拓扑排序,课程表,每日一题)