Python实现拓扑算法的示例

前言

拓扑排序是图论中一种重要的排序算法，用于对有向无环图（DAG）进行排序。在拓扑排序中，图的顶点表示任务，有向边表示任务之间的依赖关系。拓扑排序算法可以找到一种满足所有任务依赖关系的顺序。

算法原理

拓扑排序算法的基本原理如下：

• 创建一个空的排序结果列表。

• 找到图中所有入度为0的顶点（即没有依赖关系的顶点），将其加入排序结果列表。

• 移除该顶点以及与其相关的边。

• 重复步骤2和3，直到图中的所有顶点都被处理。

• 如果排序结果列表的长度等于图中的顶点数，则拓扑排序成功；否则，图中存在环，无法进行拓扑排序。

Python实现

下面是使用Python实现拓扑排序算法的示例代码：

from collections import deque
def topological_sort(graph):
? ? # 统计每个顶点的入度
? ? in_degree = {v: 0 for v in graph}
? ? # 计算每个顶点的入度
? ? for v in graph:
? ? ? ? for neighbor in graph[v]:
? ? ? ? ? ? in_degree[neighbor] += 1
? ? # 将入度为0的顶点加入队列
? ? queue = deque([v for v in graph if in_degree[v] == 0])
? ? # 保存拓扑排序的结果
? ? result = []
? ? while queue:
? ? ? ? # 取出队列中的顶点
? ? ? ? v = queue.popleft()
? ? ? ? result.append(v)
? ? ? ? # 移除顶点及其相关边
? ? ? ? for neighbor in graph[v]:
? ? ? ? ? ? in_degree[neighbor] -= 1
? ? ? ? ? ? if in_degree[neighbor] == 0:
? ? ? ? ? ? ? ? queue.append(neighbor)
? ? # 判断是否存在环
? ? if len(result) != len(graph):
? ? ? ? raise ValueError("图中存在环，无法进行拓扑排序。")
? ? return result
# 测试
graph = {
? ? 'A': ['B', 'C'],
? ? 'B': ['D'],
? ? 'C': ['D', 'E'],
? ? 'D': ['F'],
? ? 'E': ['F'],
? ? 'F': []
}
try:
? ? result = topological_sort(graph)
? ? print("拓扑排序结果:", result)
except ValueError as e:
? ? print(e)

以上代码中，graph表示图的邻接表表示法，其中每个顶点表示为一个键，其对应的值为一个列表，列表中存储了与该顶点有直接边相连的顶点。

运行代码后，将输出拓扑排序的结果。

这就是使用Python实现拓扑排序算法的示例代码。通过这个算法，我们可以对有向无环图进行

排序，找到满足任务依赖关系的顺序。

来源：https://blog.csdn.net/chy555chy/article/details/130938158