拓扑排序 python
作者:互联网
现在你总共有 numCourses
门课需要选,记为 0
到 numCourses - 1
。给你一个数组 prerequisites
,其中 prerequisites[i] = [ai, bi]
,表示在选修课程 ai
前 必须 先选修 bi
。
- 例如,想要学习课程
0
,你需要先完成课程1
,我们用一个匹配来表示:[0,1]
。
返回你为了学完所有课程所安排的学习顺序。可能会有多个正确的顺序,你只要返回 任意一种 就可以了。如果不可能完成所有课程,返回 一个空数组 。
示例 1:
输入:numCourses = 2, prerequisites = [[1,0]]
输出:[0,1]
解释:总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1] 。
示例 2:
输入:numCourses = 4, prerequisites = [[1,0],[2,0],[3,1],[3,2]] 输出:[0,2,1,3] 解释:总共有 4 门课程。要学习课程 3,你应该先完成课程 1 和课程 2。并且课程 1 和课程 2 都应该排在课程 0 之后。 因此,一个正确的课程顺序是[0,1,2,3]
。另一个正确的排序是[0,2,1,3]
。
示例 3:
输入:numCourses = 1, prerequisites = [] 输出:[0]
class Solution: def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]: #初始化一个字典,key是课程名,value都为0 courseDict={i:0 for i in range(numCourses)} #循环取列表,获取拓扑关系 for pre in prerequisites: courseDict[pre[0]] +=1 #建立一个临时列表,把入度为0的key取出来 temp=[key for key in courseDict if courseDict[key] == 0] res=[] while temp : #取temp表中的最后一个数 x=temp.pop() res.append(x) #把取出的数字对应的都-1 for pre in prerequisites: if pre[1] == x: courseDict[pre[0]] -=1 if courseDict[pre[0]] ==0: temp.append(pre[0]) #如果长度和字典长度一致就获取排序列表 if len(res) == len(courseDict): return res else: return []
标签:pre,python,courseDict,拓扑,numCourses,prerequisites,课程,key,排序 来源: https://www.cnblogs.com/yetangjian/p/15943107.html