拓扑排序 Topological Sorting

什么是拓扑排序：

和“图”相关，已经知道哪个数的后面是哪个，怎么给出正确的顺序，解可能不唯一

以上图为例，其中一个合法的顺序为：
5->4->2->3->1
注意：4和5都可以是开头，并不一定是图中箭头所指的方向，而是“入度为0”的点

具体的实现

相关Leetcode题目：剑指offerⅡ114
相关链接：什么是拓扑排序

每一次都找到入度为0的点，把他拿出来并且删除从他出去的边

from collections import defaultdict


#  edges是所有边的列表，形似[[1,2],[3,2]],1指向2，3指向2
#  e_nums,是点的总数，即一共有多少数
def topo_sort(edges, e_nums):
    # g用来存储每个点指向哪些点
    g = defaultdict(list)
    # indeg 用来记录每一个点的入度，即有几个点指向这个点
    #    只有入度为0的点才会进入排好序的队伍中，注意点的编号是从0开始的
    indeg = {c: 0 for c in range(1, e_nums + 1)}
    for i in edges:
        #  点从i[0]指向i[1]
        g[i[0]].append(i[1])
        indeg[i[1]] += 1
    #  p就是由入读为0的点组成的排好序的待输出的结果
    #  注意：p是一边遍历一边变长的
    p = [k for k, v in indeg.items() if v == 0]
    for i in p:
        for j in g[i]:
            #  将所有这个入读为0的点指向的点入度减1
            indeg[j] -= 1
            if indeg[j] == 0:
                p.append(j)
    # 如果最终的结果长度与点的个数不相同说明不存在合法的排序，返回空列表
    return p if len(p) == e_nums else []

拓扑排序的板子

特地写一个板子放在这，用来套和背

from collections import defaultdict
#  edges是所有边的列表，形似[[1,2],[3,2]],1指向2，3指向2
#  e_nums,是点的总数，即一共有多少数
def topo_sort(edges, e_nums):
    g = defaultdict(list)
    indeg = {c: 0 for c in range(1, e_nums + 1)}
    for i in edges:
        g[i[0]].append(i[1])
        indeg[i[1]] += 1
    p = [k for k, v in indeg.items() if v == 0]
    for i in p:
        for j in g[i]:
            indeg[j] -= 1
            if indeg[j] == 0:
                p.append(j)
    return p if len(p) == e_nums else []