leetcode 207. 课程表

发表于 | 分类于

现在你总共有 n 门课需要选，记为 0 到 n-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。这是不可能的。

说明:

  • 输入的先决条件是由边缘列表表示的图形，而不是邻接矩阵。详情请参见图的表示法。
  • 你可以假定输入的先决条件中没有重复的边。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import collections
class Solution(object):
    def canFinish(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: bool
        """
        def movev(v,set1,set2):
            set1.discard(v)
            set2.add(v)
            
        def dfs(v,white,grey,black):
            movev(v,white,grey)
            for i in d[v].keys():
                if i in black:
                    continue
                if i in grey:
                    return False
                if not dfs(i,white,grey,black):
                    return False
            movev(v,grey,black)
            return True
        
        d = collections.defaultdict(dict)
        for i,j in prerequisites:
            d[i][j] = 1
        white = set(range(numCourses))
        grey = set()
        black = set()
        for i in range(numCourses):
            if not dfs(i,white,grey,black):return False
        return True