Dijkstra单源最短路径python邻接表方式

下面是Dijkstra单源最短路径的Python代码，使用邻接表方式实现：

from heapq import heappush, heappop

def dijkstra(graph, start):
    # 初始化距离字典和堆
    dist = {node: float('inf') for node in graph}
    dist[start] = 0
    heap = [(0, start)]

    while heap:
        # 弹出堆顶元素
        (distance, current_node) = heappop(heap)

        # 如果当前节点已经被访问过，跳过
        if distance > dist[current_node]:
            continue

        # 遍历当前节点的邻居节点
        for neighbor, weight in graph[current_node].items():
            # 计算到邻居节点的距离
            distance_to_neighbor = distance + weight

            # 如果从当前节点到邻居节点的距离更短，则更新距离字典和堆
            if distance_to_neighbor < dist[neighbor]:
                dist[neighbor] = distance_to_neighbor
                heappush(heap, (distance_to_neighbor, neighbor))

    return dist

其中，`graph` 是一个邻接表，表示图的结构，`start` 是起点。函数返回一个字典，表示从起点到每个节点的最短距离。注意，这里使用了 Python 内置的 `heapq` 模块，它提供了堆的实现，可以高效地处理 Dijkstra 算法中的堆操作。