[LeetCode]207 课程表

题目描述

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

完成课程 1 ，我们用一个匹配来表示他们: [0,1]
给定课程总量以及它们的先决条件，判断是否可能完成所有课程的学习？

示例1:

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

示例2:

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

说明:

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

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution:
    def canFinish(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: bool
        """
        g = collections.defaultdict(list)
        for u,v in prerequisites:
            g[v].append(u)
        visited=[0]*numCourses
        def dfs(i):
            if visited[i]==1:return False
            if visited[i]==2:return True
            visited[i]=1
            for j in g[i]:
                if not dfs(j):return False
            visited[i]=2
            return True
        for i in range(numCourses):
            if not dfs(i):return False
        return True