Python-Bellman-Ford算法

Bellman-Ford算法是一种用于解决单源最短路径问题的算法，它可以处理带有负权边和负权回路的图。它的主要思想是通过对所有边进行多轮松弛操作来逐步更新顶点的最短路径估计值。
算法的步骤如下：

1.初始化距离数组，将起始点到其他顶点的距离设为无穷大，起始点的距离设为0。
2.对所有边进行n-1轮松弛操作，其中n是图中顶点的数量。每轮松弛操作都会更新顶点的最短路径估计值。
3.检查是否存在负权回路。如果在第n-1轮松弛操作后，仍然可以通过松弛操作减小某些顶点的最短路径估计值，则说明存在负权回路。
4.如果存在负权回路，则算法无法得出正确的最短路径。否则，最终得到的距离数组就是起始点到其他顶点的最短路径。

Bellman-Ford算法的时间复杂度为O(VE)，其中V是顶点数，E是边数。这是因为算法需要对所有边进行n-1轮松弛操作。
需要注意的是，Bellman-Ford算法不能处理存在负权回路的情况，因为在负权回路中，没有最短路径的概念。如果存在负权回路，则算法会进入无限循环。

# 定义一个函数来实现Bellman-Ford算法
def bellman_ford(graph, start):
    # 初始化距离字典，将起始点到其他顶点的距离设为无穷大，起始点的距离设为0
    distance = {vertex: float('inf') for vertex in graph}
    distance[start] = 0

    # 对所有边进行n-1轮松弛操作，其中n是图中顶点的数量
    for _ in range(len(graph) - 1):
        for vertex in graph:
            for neighbor in graph[vertex]:
                # 如果通过当前顶点可以获得更短的路径，则更新最短路径估计值
                if distance[vertex] + graph[vertex][neighbor] < distance[neighbor]:
                    distance[neighbor] = distance[vertex] + graph[vertex][neighbor]

    # 检查是否存在负权回路
    for vertex in graph:
        for neighbor in graph[vertex]:
            # 如果在第n-1轮松弛操作后，仍然可以通过松弛操作减小某些顶点的最短路径估计值，则说明存在负权回路
            if distance[vertex] + graph[vertex][neighbor] < distance[neighbor]:
                print("存在负权回路")
                return

    # 返回起始点到其他顶点的最短路径字典
    return distance

# 测试代码
if __name__ == "__main__":
    # 定义一个有向加权图的邻接表表示
    graph = {
        'A': {'B': 6, 'D': 1},
        'B': {'C': 5},
        'C': {'B': -2},
        'D': {'B': 2, 'C': 4}
    }

    start_vertex = 'A'
    shortest_distances = bellman_ford(graph, start_vertex)

    # 打印起始点到其他顶点的最短路径
    for vertex in shortest_distances:
        print(f"起始点到顶点 {vertex} 的最短路径距离为 {shortest_distances[vertex]}")