Question

给定列表：[1,5,6]，[2,3,5,6]，[2,5]等（不一定按任何排序顺序），如果x在一个列表中先于y，则x在x和y的每个列表中都先于y，我想找到拓扑排序的所有元素的列表（如果x在任何其他列表中的y前面，则x在此列表中的y前面。）可能有许多解决方案，其中我希望他们中的任何一个。

在Python中实现此功能的最简单方法是什么。

Answer 1

这是@ unutbu的networkx解决方案的一个稍微简单的版本：

import networkx as nx
data=[[1, 5, 6], [2, 3, 5, 6], [2, 5], [7]]
G = nx.DiGraph()
for path in data:
    G.add_nodes_from(path)
    G.add_path(path)
ts=nx.topological_sort(G)
print(ts)
# [7, 2, 3, 1, 5, 6]

Answer 2

使用networkx，特别是networkx.topological_sort：

import networkx as nx

data=[[1, 5, 6], [2, 3, 5, 6], [2, 5], [7]]
G=nx.DiGraph()
for row in data:
    if len(row)==1:
        G.add_node(row[0])
    else:
        for v,w in (row[i:i+2] for i in xrange(0, len(row)-1)):
            G.add_edge(v,w)


ts=nx.topological_sort(G)
print(ts)
# [2, 3, 1, 5, 6]

Answer 3

我的解决方案（使用@unutbu中的一些代码）

import collections

retval = []
data = [[1,2,3], [4,5,6], [2, 5], [3, 6], [1, 7]]
in_edges = collections.defaultdict(set)
out_edges = collections.defaultdict(set)
vertices = set()
for row in data:
    vertices |= set(row)
    while len(row) >= 2:
        w = row.pop()
        v = row[-1]
        in_edges[w].add(v)
        out_edges[v].add(w)
def process(k):
    vertices.remove(k)
    retval.append(k)
    for target in out_edges[k]:
        in_edges[target].remove(k)
    for target in out_edges[k]:
        if not in_edges[target]:
            process(target)
    out_edges[k] = set()

while vertices:  # ideally, this should be a heap
    for k in vertices:
        if not in_edges[k]:
            process(k)
            break

print(retval)

使用指示边的列表进行Python拓扑排序

3 个答案: