题目描述
你这个学期必须选修 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]]))