Passo 9

Instruções

À medida que o algoritmo é executado, você precisa acompanhar quais nós você já visitou para não processá-los mais de uma vez. Para fazer isso, crie uma lista chamada visited e inicialize-a com False para cada nó.

O que fazer:

Testes:

Você deve ter uma lista chamada `visited` dentro da função `shortest_path`.
Você deve inicializar a lista `visited` com `False` para cada nó usando `[False] * n`.

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)]

Preview