Passo 25

Instruções

Finalmente, imprima os resultados para o nó atual. Sua saída deve mostrar duas coisas, a distância do start_node até node_no e o caminho percorrido para chegar lá. Use um f-string para formatar a saída para mostrar tanto a distância e o caminho completo usando o formato:

\n[starting node]-[node number] distance: [distance]\nPath: [path]

Onde:

[starting node] é o start_node
[node number] é o node_no
[distance] é a distância do start_node até node_no
[path] é o caminho percorrido para chegar ao node_no

O que fazer:

Testes:

Você deve imprimir a saída correta usando `print(f'\n{start_node}-{node_no} distance: {distances[node_no]}\nPath: {path}')`.

INF = float('inf')
adj_matrix = [
    [0, 5, 3, INF, 11, INF],
    [5, 0, 1, INF, INF, 2],
    [3, 1, 0, 1, 5, INF],
    [INF, INF, 1, 0, 9, 3],
    [11, INF, 5, 9, 0, INF],
    [INF, 2, INF, 3, INF, 0],
]

def shortest_path(matrix, start_node, target_node=None):
    n = len(matrix)
    distances = [INF] * n
    distances[start_node] = 0
    paths = [[node_no] for node_no in range(n)]
    visited = [False] * n

    for _ in range(n):
        min_distance = INF
        current = -1
        for node_no in range(n):
            if not visited[node_no] and distances[node_no] < min_distance:
                min_distance = distances[node_no]
                current = node_no

        if current == -1:
            break
        visited[current] = True

        for node_no in range(n):
            distance = matrix[current][node_no]
            if distance != INF and not visited[node_no]:
                new_distance = distances[current] + distance
                if new_distance < distances[node_no]:
                    distances[node_no] = new_distance
                    paths[node_no] = paths[current] + [node_no]


    targets = [target_node] if target_node is not None else range(n)
    for node_no in targets:
        if node_no == start_node or distances[node_no] == INF:
            continue
        string_path = (str(n) for n in paths[node_no])
        path = ' -> '.join(string_path)

Preview