Leetcode C++《热题 Hot 100-65》207. 课程表

Leetcode C++《热题 Hot 100-65》207. 课程表

1. 题目

你这个学期必须选修 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 <= numCourses <= 10^5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/course-schedule
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 思路

• 拓扑排序，用出入度方法容易理解hh，熟悉的配方。[0, 1], 先学习1才能学习0，那么0的入度为++；学习1之后要把它学了之后能学的课程入度都–，需要一个map
• 如果不能完美的进行拓扑排序，返回false

3. 代码

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        //拓扑排序，用出入度方法容易理解hh，熟悉的配方
        // [0, 1], 先学习1才能学习0，那么0的入度为++；学习1之后要把它学了之后能学的课程入度都--，需要一个map
        map<int, int> degrees;
        map<int, vector<int>> relations;
        queue<int> temp;
        map<int, int>:: iterator it_degree;
        map<int, vector<int>>:: iterator it_relation;

        for (int i = 0; i < prerequisites.size(); i++) {
            vector<int> relation = prerequisites[i];
            it_degree = degrees.find(relation[0]);
            if (it_degree == degrees.end()) {
                degrees.insert(make_pair(relation[0], 1));
            } else {
                degrees[relation[0]]++;
            }
            it_relation = relations.find(relation[1]);
            if (it_relation == relations.end()) {
                vector<int> childrens;
                childrens.push_back(relation[0]);
                relations.insert(make_pair(relation[1], childrens));
            } else {
                relations[relation[1]].push_back(relation[0]);
            }
        }

        //找到入度为0的点，放进temp
        vector<int> find_zero;
        for (int i = 0; i < numCourses; i++) {
            it_degree = degrees.find(i);
            if (it_degree == degrees.end())
                temp.push(i);
        }
        /*for (it_degree = degrees.begin(); it_degree != degrees.end(); it_degree++) {
            cout << it_degree->first << " " << it_degree->second << endl;
        }*/

        int nums = 0;
        while (!temp.empty()) {
            int one_lesson = temp.front();
            temp.pop();
            nums++;
            it_relation = relations.find(one_lesson);
            if (it_relation != relations.end()) {
                vector<int> childrens_lesson = relations[one_lesson];
                for (int i = 0; i < childrens_lesson.size(); i++) {
                    degrees[childrens_lesson[i]]--;
                    if (degrees[childrens_lesson[i]] == 0) {
                        temp.push(childrens_lesson[i]);
                    }
                    if (degrees[childrens_lesson[i]] < 0) {
                       return false;
                    }
                }
            }
        }
        if (nums == numCourses)
            return true;
        return false;

    }
};