python求最短路径
时间: 2023-08-31 21:13:01 浏览: 39
Python中求最短路径的方法有很多,其中一种比较常用的方法是使用Dijkstra算法。Dijkstra算法是一种贪心算法,用于解决带有非负权值边的最短路径问题。以下是一个使用Dijkstra算法的示例代码:
```python
import heapq
def dijkstra(graph, start):
distances = {node: float('inf') for node in graph}
distances[start] = 0
queue = [(0, start)]
visited = set()
while queue:
curr_distance, curr_node = heapq.heappop(queue)
if curr_node in visited:
continue
visited.add(curr_node)
for neighbor, weight in graph[curr_node].items():
distance = curr_distance + weight
if distance < distances[neighbor]:
distances[neighbor] = distance
heapq.heappush(queue, (distance, neighbor))
return distances
```
在这个示例代码中,使用了一个字典来表示图,其中字典的键表示节点,字典的值也是一个字典,用来表示该节点到它的邻居节点的边的权值。函数的参数`start`表示起始节点。
该函数返回一个字典,表示从起始节点到每个节点的最短距离。