Leetcode.0207 | 课程表

题目

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

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

例如，先修课程对 [0, 1] 表示：想要学习课程 0 ，你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习？如果可以，返回 true ；否则，返回 false 。

示例

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

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

解决方法

使用拓扑排序解决，将每一门课视为一个节点。如果课程A是课程B的前置课程，则在A->B出连一条有向边。使用广度优先遍历判断这张图是否有环，如果有的话就说明不可能完成。

广度优先搜索借助队列来进行。初始时，所有入度为0的节点都被放入队列中，它们就是可以作为拓扑排序最前面的节点，并且它们之间的相对顺序是无关紧要的。

在广度优先搜索的每一步中，首先取出队列头节点u，将u放入结果数组res中。然后移除u的所有出边，将u的所有相邻节点的入度减一。如果减了以后u的某个邻接节点的入度变为0，则将这个节点放入队列中。

在广度优先搜索的过程结束后，如果答案中包含了所有的节点，则找到了一种拓扑排序，否则说明图中存在环。

代码实现

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        if(prerequisites.size() <= 1) return true;
        vector<int> inDegree(numCourses,0); //保存各个节点的入度
        queue<int> data;
        vector<int> res;
        for(vector<int>& temp:prerequisites){ //统计各个节点的入度
            inDegree[temp[0]] ++;  //在拓扑图中的边为temp[1]->temp[0],temp[0]的入度+1.
        }
        for(int i = 0;i < numCourses;i ++){//寻找入度为０的元素，入队
            if(inDegree[i] == 0){
                data.push(i);
            }
        }
        while(!data.empty()){
            int top = data.front(); //取出队列中的第一个元素。
            res.push_back(top);
            data.pop();
            for(vector<int>& temp:prerequisites){ //依次遍历
                if(temp[1] == top)  // 如果遇到队列中第一个元素的邻接节点
                {
                    inDegree[temp[0]] --; //将邻接节点的入度减一
                    if(inDegree[temp[0]] == 0){
                        data.push(temp[0]);
                    }
                }
            }
        }
        return res.size() == numCourses;
    }
};

备注

C++中queue的用法

• front()：返回 queue 中第一个元素的引用。如果 queue 是常量，就返回一个常引用；如果 queue 为空，返回值是未定义的。
• back()：返回 queue 中最后一个元素的引用。如果 queue 是常量，就返回一个常引用；如果 queue 为空，返回值是未定义的。
• push(const T& obj)：在 queue 的尾部添加一个元素的副本。这是通过调用底层容器的成员函数 push_back() 来完成的。
• push(T&& obj)：以移动的方式在 queue 的尾部添加元素。这是通过调用底层容器的具有右值引用参数的成员函数 push_back() 来完成的。
• pop()：删除 queue 中的第一个元素。
• size()：返回 queue 中元素的个数。
• empty()：如果 queue 中没有元素的话，返回 true。
• emplace()：用传给 emplace() 的参数调用 T 的构造函数，在 queue 的尾部生成对象。
• swap(queue<T> &other_q)：将当前 queue 中的元素和参数 queue 中的元素交换。它们需要包含相同类型的元素。也可以调用全局函数模板 swap() 来完成同样的操作。