力扣—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。这是不可能的。

解题思路

拓扑排序，使用队列，首先选择入度为0的节点入队。依次从队列中弹出元素，并把以该节点为前驱节点的后继节点入度减1，因此需要维护一个入度表和邻接表，每弹出一个节点，courses的值减1，最后判断courses值是否为0。

class Solution(object):
    def canFinish(self, numCourses, prerequisites):
        queue = []
        indegrees = [0 for _ in range(numCourses)]     #入度表
        adj = [[] for _ in range(numCourses)]         #邻接表
        for item in prerequisites:
            indegrees[item[1]] += 1
            adj[item[0]].append(item[1])
        for i in range(len(indegrees)):
            if not indegrees[i]:
                queue.append(i)
        while queue:
            course = queue.pop(0)
            numCourses -= 1
            for temp in adj[course]:
                indegrees[temp] -= 1
                if not indegrees[temp]:
                    queue.append(temp)
        return numCourses == 0

S = Solution()
print(S.canFinish(2, [[1,0], [0, 1]]))