Dijkstra最短路径算法 代码

以下是Dijkstra最短路径算法的Python实现：

import heapq

def dijkstra(graph, start):
    # 初始化距离字典和堆
    distances = {node: float('inf') for node in graph}
    distances[start] = 0
    heap = [(0, start)]
    
    # 开始遍历
    while heap:
        (current_distance, current_node) = heapq.heappop(heap)
        
        # 如果当前节点已经被遍历过，就跳过
        if current_distance > distances[current_node]:
            continue
        
        # 遍历当前节点的邻居
        for neighbor, weight in graph[current_node].items():
            distance = current_distance + weight
            
            # 如果发现更短的路径，就更新距离字典和堆
            if distance < distances[neighbor]:
                distances[neighbor] = distance
                heapq.heappush(heap, (distance, neighbor))
    
    return distances

其中，`graph`是以字典形式表示的图，`start`是起点。`graph`的结构为：

{
    'A': {'B': 5, 'C': 1},
    'B': {'A': 5, 'C': 2, 'D': 1},
    'C': {'A': 1, 'B': 2, 'D': 4, 'E': 8},
    'D': {'B': 1, 'C': 4, 'E': 3, 'F': 6},
    'E': {'C': 8, 'D': 3},
    'F': {'D': 6}
}

其中，每个键值对表示一个节点和它的邻居节点。节点名字用字符串表示，邻居节点用字典表示，键是邻居节点的名字，值是到邻居节点的距离。