使用桶排序算法中的kahn(卡恩)算法,也可以使用dfs。
这里使用卡恩算法,主要维护一个列表cnt,cnt【i】表示能到达节点i的边,比如说:a到c有一条边,b到c有一条边,那么cnt【c】= 2。当然,如果从a走到c一次,cnt【i】= 1。如果b再到cc一次,那么cnt【i】= 0。如果某个节点i,cnt【i】= 0,说明没有边能到达i节点,就是说i节点前面没有其他节点了,那么可以选择i节点。
class Solution:def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:ans = []# kahn 算法edge = defaultdict(list)cnt = [0]*numCoursesfor a, b in prerequisites:edge[b].append(a)cnt[a] += 1qu = deque()for i in range(numCourses):if cnt[i] == 0:qu.append(i)while qu:f = qu.popleft()ans.append(f)for nex in edge[f]:cnt[nex] -= 1if cnt[nex] == 0:qu.append(nex)return ans if len(ans) == numCourses else []