Dijkstra算法是一种用于计算单源最短路径的经典算法。最小权重Dijkstra树是指从源节点到所有其他节点的最短路径树,其中每条边的权重是正数。
Dijkstra算法主要分为两种实现方式:
原因:Dijkstra算法基于贪心策略,每次选择当前距离最小的节点进行扩展。如果图中存在负权重边,可能会导致算法无法正确找到最短路径。
解决方法:使用Bellman-Ford算法或SPFA(Shortest Path Faster Algorithm)来处理包含负权重边的图。
原因:在稀疏图上,使用数组实现的Dijkstra算法时间复杂度较高,为O(V^2)。优先队列实现的时间复杂度为O((V + E) log V),性能较好。
解决方法:在稀疏图上,使用优先队列实现Dijkstra算法。
import heapq
def dijkstra(graph, start):
queue = []
heapq.heappush(queue, (0, start))
distances = {node: float('inf') for node in graph}
distances[start] = 0
shortest_path = {}
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))
shortest_path[neighbor] = current_node
return distances, shortest_path
# 示例图
graph = {
'A': {'B': 1, 'C': 4},
'B': {'A': 1, 'C': 2, 'D': 5},
'C': {'A': 4, 'B': 2, 'D': 1},
'D': {'B': 5, 'C': 1}
}
distances, shortest_path = dijkstra(graph, 'A')
print("Distances:", distances)
print("Shortest Path:", shortest_path)
领取专属 10元无门槛券
手把手带您无忧上云