Leetcode 207. 课程表

https://leetcode.cn/problems/course-schedule/

你这个学期必须选修 numCourses 门课程,记为 0 到 numCourses - 1 。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出,
其中 prerequisites[i] = [ai, bi] ,表示如果要学习课程 ai 则 必须 先学习课程  bi 。
例如,先修课程对 [0, 1] 表示:想要学习课程 0 ,你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习?如果可以,返回 true ;否则,返回 false 。


示例 1:
输入:numCourses = 2, prerequisites = [[1,0]]
输出:true
解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。

示例 2:
输入:numCourses = 2, prerequisites = [[1,0],[0,1]]
输出:false
解释:总共有 2 门课程。学习课程 1 之前,你需要先完成​课程 0 ;并且学习课程 0 之前,你还应先完成课程 1 。这是不可能的。
 

提示:
1 <= numCourses <= 105
0 <= prerequisites.length <= 5000
prerequisites[i].length == 2
0 <= ai, bi < numCourses
prerequisites[i] 中的所有课程对 互不相同

解法
拓扑图

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        vector<unordered_set<int>> gra(numCourses+10);
        vector<int>  inDegree(numCourses + 10);
        for (int i = 0; i < prerequisites.size(); i++) {
            gra[prerequisites[i][1]].insert(prerequisites[i][0]);
            inDegree[prerequisites[i][0]]++;
        }
        queue<int> q;
        for (int i = 0; i < numCourses; i++) {
            if (inDegree[i] == 0) {
                q.push(i);
            }
        }
        int popcnt =0;
        while (!q.empty()) {
            int p = q.front(); q.pop();popcnt++;
            if(popcnt> numCourses) return false;

            for (auto& e : gra[p]) {
                inDegree[e]--;
                if (inDegree[e] == 0) {
                    q.push(e);
                }
            }

            gra[p].clear();
        }

        for (int i = 0; i < numCourses; i++) {
            if (inDegree[i] > 0) return false;
        }

        return true;
    }
};

我的视频空间

posted on 2022-10-02 22:42  itdef  阅读(77)  评论(0编辑  收藏  举报

导航