您的位置：首页 > 编程语言 > Python开发

dijkstra算法的python实现

2018-03-29 23:01 423 查看

def dijkstra(graph,src):
    # 判断图是否为空，如果为空直接退出
    if graph is None:
        return None
    nodes = [i for i in range(len(graph))] # 获取图中所有节点
    visited=[] # 表示已经路由到最短路径的节点集合
    if src in nodes:
        visited.append(src)
        nodes.remove(src)
    else:
        return None
    distance={src:0} # 记录源节点到各个节点的距离
    for i in nodes:
        distance[i]=graph[src][i] # 初始化
    # print(distance)
    path={src:{src:[]}} # 记录源节点到每个节点的路径
    k=pre=src
    while nodes:
        mid_distance=float('inf')
        for v in visited:
            for d in nodes:
                new_distance = graph[src][v]+graph[v][d]
                if new_distance < mid_distance:
                    mid_distance=new_distance
                    graph[src][d]=new_distance # 进行距离更新
                    k=d
                    pre=v
        distance[k]=mid_distance # 最短路径
        path[src][k]=[i for i in path[src][pre]]
        path[src][k].append(k)
        # 更新两个节点集合
        visited.append(k)
        nodes.remove(k)
       # print(visited,nodes) # 输出节点的添加过程
    return distance,path

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签： python dijkstra

相关文章推荐

新的分享

章节导航