写一个迪杰斯特拉的算法
时间: 2023-04-10 17:04:45 浏览: 108
好的,以下是迪杰斯特拉算法的代码:
```python
import heapq
def dijkstra(graph, start):
distances = {node: float('inf') for node in graph}
distances[start] = 0
pq = [(0, start)]
while pq:
current_distance, current_node = heapq.heappop(pq)
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(pq, (distance, neighbor))
return distances
```
这是一个 Python 实现的迪杰斯特拉算法,它接受一个图和一个起点作为输入,并返回从起点到每个节点的最短距离。算法使用堆优化的 Dijkstra 算法来实现,时间复杂度为 O(E log V),其中 E 是边数,V 是节点数。
阅读全文