LeetCode-207-课程表

LeetCode-207-课程表

题目

你这个学期必须选修 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。这是不可能的。

思路

一开始只是觉得有点眼熟，但是没有往图的方向想；

忍不住看了一下题解，发现这道题原来是拓扑排序，难怪那么眼熟，是以前做过的；

代码中用队列存放的数字，其实用栈或者链表都可以的，代码如下：

代码

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        typedef vector<int> vec;
        typedef vector<vec> matvec;

        // 边界检查
        if (numCourses <= 0) return true;

        // 构建图
        matvec graph(numCourses);
        vec indegree(numCourses);
        for (vec &v:prerequisites) {
            ++indegree[v[0]];
            graph[v[1]].push_back(v[0]);
        }

        // 删除入度为0的点
        queue<int> myque;
        for (int i=0; i<numCourses; i++) {
            if (indegree[i] == 0) myque.push(i);
        }

        while (!myque.empty()) {
            int val = myque.front(); myque.pop();
            //cout<<val<<endl;
            for (int &x:graph[val]) {
                --indegree[x];
                if (indegree[x] == 0) myque.push(x);
            }
        }
        for (int &x:indegree) {
            if (x!=0) return false;
        }
        return true;
    }
};