python--leetcode207. 课程表

现在你总共有 n 门课需要选，记为 0 到 n - 1。

一些课程在修之前需要先修另外的一些课程，比如要学习课程 0 你需要先学习课程 1 ，表示为: [0,1]
给定 n 门课以及他们的先决条件，判断是否可能完成所有课程？

例如：

2, [[1,0]]

总共有2门课程可供选择。选取课程 1之前，你需要完成课程 0。所以这是可能的。

2, [[1,0],[0,1]]

总共有 2 门课程可供选择。要参加课程 1，你需要先完成​课程 0，并且选取课程 0之前，你还应先完成课程 1。所以这是不可能的。

注意:

1. 输入先决条件是由边缘列表表示的图形，而不是邻接矩阵。详细了解 图的表示法。
2. 你可以假定输入先决条件中没有重复的边。

提示:

1. 这个问题相当于查找一个循环是否存在于有向图中。如果存在循环，则不存在拓扑排序，因此不可能采取所有课程。
2. 通过 DFS 进行拓扑排序 - 一个关于Coursera的精彩视频教程（21分钟），介绍拓扑排序的基本概念。

3. 拓扑排序也可以通过 BFS 完成。

这一题就是拓扑排序，拓扑排序的原理见我上一篇博客，直接上代码吧，注释中有讲解：

class Solution:
    def canFinish(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: bool
        """
        dic={}
        for i in range(numCourses):
            dic[i]=0   #初始化所有结点入度
        for i in range(len(prerequisites)):
            head=prerequisites[i][1]
            tail=prerequisites[i][0]
            dic[tail]+=1      #根据边的关系得到入度
        from queue import  Queue
        q=Queue()
        result=[]
        for i in range(len(dic)):
            if(dic[i]==0):
                q.put(i)   #将入度为0的点加进队列
        while(q.qsize()!=0):
            node=q.get()
            result.append(node)
            for j in range(len(prerequisites)):
                if(prerequisites[j][1]==node):
                    tail2=prerequisites[j][0]
                    dic[tail2]-=1   #将node结点从图中删除
                    if(dic[tail2]==0):
                        q.put(tail2)
        if(len(result)==numCourses):return True
        else :return False




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