Leetcode-图论

Leetcode-图论

图论

207. 课程表

你这个学期必须选修 numCourses 门课程，记为 0 到 numCourses - 1 。

在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出，其中 prerequisites[i] = [a<sub>i</sub>, b<sub>i</sub>] ，表示如果要学习课程 a<sub>i</sub> 则 必须 先学习课程 b<sub>i</sub> ~ ~ 。

• 例如，先修课程对 [0, 1] 表示：想要学习课程 0 ，你需要先完成课程 1 。

请你判断是否可能完成所有课程的学习？如果可以，返回 true ；否则，返回 false 。

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        vector<int> totalnum(numCourses,0);
        vector<vector<int> > matrix(numCourses, vector<int>(0,0));
        for(int i=0;i<prerequisites.size();i++){
            totalnum[prerequisites[i][0]] += 1;
            matrix[prerequisites[i][1]].push_back(prerequisites[i][0]);
        }
        bool judge = true;
        while(judge){
            judge = false;
            for(int i=0;i<numCourses;i++){
                if(totalnum[i] == 0){
                    judge = true;
                    for(int j=0;j<matrix[i].size();j++){
                        totalnum[matrix[i][j]] -= 1;
                    }
                    totalnum[i] = -1;
                }
            }
        }
        for(int i=0;i<numCourses;i++){
            if(totalnum[i] != -1){
                return false;
            }
        }
        return true;
    }
};