Dijkstra算法的python表现
时间: 2024-05-18 20:09:56 浏览: 61
Dijkstra算法是一种用于图中单源最短路径的算法,它通过不断更新起始点到各个顶点的距离来寻找最短路径。以下是Dijkstra算法在Python中的实现:
```python
import heapq
def dijkstra(graph, start):
distances = {node: float('inf') for node in graph} # 初始化距离字典,将起始点到各点的距离都设为无穷大
distances[start] = 0 # 将起始点到自身的距离设为0
heap = [(0, start)] # 将起始点加入堆中
while heap:
(dist, current_node) = heapq.heappop(heap) # 取出堆中距离最短的节点
if dist > distances[current_node]: # 如果取出节点到起始点的距离已经大于之前更新过的距离,就忽略该节点
continue
for neighbor, weight in graph[current_node].items(): # 遍历当前节点的所有邻居节点
distance = dist + weight # 计算当前节点到邻居节点的距离
if distance < distances[neighbor]: # 如果当前节点到邻居节点的距离比之前更新过的距离更短,就更新距离字典,并将邻居节点加入堆中
distances[neighbor] = distance
heapq.heappush(heap, (distance, neighbor))
return distances
```
以上代码中,`graph`是一个字典,它表示图中各个节点之间的关系和权值。例如,如果有一条从节点A到节点B的边权为2的边,则可以表示为`graph = {'A': {'B': 2}}`。`start`表示起始节点。
下面是一些相关问题:
阅读全文