1. 问题描述:
你这个学期必须选修 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。这是不可能的。
提示:
输入的先决条件是由边缘列表表示的图形,而不是邻接矩阵 。详情请参见图的表示法。
你可以假定输入的先决条件中没有重复的边。
1 <= numCourses <= 10 ^ 5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/course-schedule
2. 思路分析:
这道题目与力扣第207道题目是类似的,属于拓扑排序的模板题,我们只需要在判断判断拓扑排序的过程中记录下入度为0的节点即可。拓扑排序一般有以下几个步骤:
① 根据节点的之间的关系创建有向图,因为使用的是python语言所以使用字典来创建有向图
② 使用数组或者列表来统计节点的入队,可以在①中根据节点之间的联系统计节点的入度
③ 声明一个队列,将入度为0的节点加入到队列中,python语言可以使用list列表来模拟队列,或者是collections.deque()声明一个双端队列
④ 当队列不为空的时候执行循环,弹出队首节点,遍历弹出节点的连接的边,删除掉边连接节点的入度,当入队为0的时候加入到队列中
⑤ 在弹出队列节点的时候进行计数,最后判断计数的值是否等于所以节点的值,满足则存在拓扑排序(如果存在环那么所有节点的入度肯定不为0)不满足返回None
3. 代码如下:
from typing import List
import collections
class Solution:
def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:
dic = collections.defaultdict(list)
d = [0] * numCourses
for c in prerequisites:
dic[c[1]].append(c[0])
d[c[0]] += 1
queue = list()
# 将入度为0的节点入队
for i in range(numCourses):
if d[i] == 0:
queue.append(i)
count = 0
res = list()
while queue:
# 弹出队首节点
poll = queue.pop(0)
# 遍历当前节点连接的其他节点
next = dic[poll]
res.append(poll)
count += 1
for i in next:
# 对应的节点的度数减1
d[i] -= 1
# 将入度为0的节点入队
if d[i] == 0:
queue.append(i)
# 判断是否存在拓扑排序
if count != numCourses: res = list()
return res