207. Course Schedule

最新推荐文章于 2021-12-10 13:25:51 发布

weixin_34006468

最新推荐文章于 2021-12-10 13:25:51 发布

阅读量47

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/JTechRoad/p/9001297.html

本文介绍了一种使用深度优先搜索（DFS）和拓扑排序的方法来解决课程安排问题。通过定义节点的三种状态（未访问、正在访问、已访问），可以有效地检测到图中是否存在环，进而判断是否能完成所有课程的学习。

https://leetcode.com/problems/course-schedule/description/

DFS, detect cycle. For each node, we have 3 states during DFS: 1) un-visited, 2) visiting, 3) visited.

class Solution {
public:
    bool canFinish(int n, vector<pair<int, int>>& p) {
        vector<int> visited(n, -1);     // un-visited
        vector<set<int>> v(n, set<int>());
        for (int i = 0; i < p.size(); i++) {
            v[p[i].second].insert(p[i].first);
        }
        for (int i = 0; i < n; i++)
            if (visited[i] == -1 && dfs(v, visited, i))
                return false;
        return true;
    }
    bool dfs(vector<set<int>>& v, vector<int>& visited, int n) {
        visited[n] = 1;     // visiting
        for (auto nb : v[n]) {
            if (visited[nb] == 1)
                return true;
            else if (visited[nb] == 2)
                continue;
            else {
                if (dfs(v, visited, nb))
                    return true;
            }
        }
        visited[n] = 2;     // visited
        return false;
    }
};