一起来刷题系列之拓扑排序

课程表 II

现在你总共有 numCourses 门课需要选，记为 0 到 numCourses - 1。给你一个数组 prerequisites ，其中 prerequisites[i] = [ai, bi] ，表示在选修课程 ai 前 必须 先选修 bi 。

例如，想要学习课程 0 ，你需要先完成课程 1 ，我们用一个匹配来表示：[0,1] 。
返回你为了学完所有课程所安排的学习顺序。可能会有多个正确的顺序，你只要返回 任意一种 就可以了。如果不可能完成所有课程，返回 一个空数组 。

输入：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] 。

https://leetcode-cn.com/problems/course-schedule-ii

分析

graph的一维下标代表每个节点，对应下标保存[ ]数组，也就是保存下一个节点（箭头朝向节点）
flagin保存每个节点的入度，从入度为0 的节点开始遍历，每访问一个节点，它所链接的所有下一个节点的入度-1，直到所有入度为0的节点都被访问，结果保存在ans数组中。
如果ans的长度和numCourses一样，直接返回ans；
如果不一样，返回[],因为存在环

拓扑排序广度优先遍历实现

class Solution:
    def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:
        # 使用广度优先搜索
        graph =[[] for i in range(numCourses)]
        flagin=[0 for i in range(numCourses)]

        for item in prerequisites:
            pre = item[1]
            nxt = item[0]

            graph[pre].append(nxt)

            flagin[nxt]+=1
        
        order = []
        for i in range(numCourses):
            if flagin[i]==0:
                order.append(i)
            
        ans = []
        while len(order)>0:
            
            fir = order.pop(0)
            ans.append(fir)

            tmp = graph[fir]
            for it in tmp:
                flagin[it]-=1
                if flagin[it]==0:
                    order.append(it)

        if len(ans)!=numCourses:
            return []
        return ans

拓扑排序之深度优先遍历

  # 拓扑排序

        data = [[] for i in range(numCourses)]
        # print(data)
        # 保存入度
        flag = [0 for i in range(numCourses)]

        for item in prerequisites:
            pre = item[1]
            last = item[0]
            flag[last]+=1
            data[pre].append(last)
        # print(data,flag)

        cla = []

        def solve(cur):
            cla.append(cur)
            flag[cur]=998
            # 入度-1
            tmp = data[cur]
            for item in tmp:
                flag[item]-=1
                if flag[item]==0:
                    solve(item)




        for i in range(numCourses):
            if flag[i]==0:
                solve(i)
        
        for i in range(numCourses):
            if flag[i]!=998:
                return []
        
        return cla