Dijkstra方法的代码实现 算法实现(Python):
时间: 2023-08-14 20:20:23 浏览: 101
Dijkstra(迪杰斯特拉)算法实现
4星 · 用户满意度95%
好的,以下是Dijkstra算法的Python代码实现:
```python
import heapq
def dijkstra(graph, start):
distances = {node: float('inf') for node in graph}
distances[start] = 0
queue = [(0, start)]
while queue:
current_distance, current_node = heapq.heappop(queue)
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(queue, (distance, neighbor))
return distances
```
其中,`graph` 是一个字典,表示图的邻接表,`start` 表示起点。这个算法使用了堆(heap)来优化时间复杂度,可以更快地找到下一个距离最短的节点。最后,返回一个字典 `distances`,表示每个节点到起点的最短距离。
阅读全文