网上关于Dijkstra算法与Floyd算法的python实现各种各样,没有一个简单的介绍,下面用精简的python代码实现了这两种方法:
Dijkstra(迪杰斯特拉)可以求一个结点到其他结点的最短路径,且所有边不能有负值
Floyd(弗洛伊德)直接求出所有节点到其他结点的最短路径,边可以为负值,但不能有负权回路
Dijkstra算法核心步骤:
- 初始化两个集合,S集合初始时 只有当前要计算的节点,
A->A
,Q集合初始时为 除A以外所有结点 - 通过S中已经确定的点比较Q中点的直达距离与中转距离,并将所有更短路径保存在dist中
- 将最短路径结点加入S集合,并从Q中移除
def dijkstra(graph,s):
if graph is None:
return None
# 记录已访问结点
S = [s]
# 记录未访问结点
Q = [i for i in range(len(graph))]
Q.remove(s)
dist = [i for i in graph[s]]
# 得到下个路径最近结点
next = -1
while Q:
mid_distance = float("inf")
# 用已知结点中转
for i in S:
# 得到未确定的最短路径结点
for j in Q:
if dist[i] + graph[i][j] < mid_distance: