Cod sursa(job #2449063)

Utilizator voyagerSachelarie Bogdan voyager Data 18 august 2019 00:41:51
Problema Floyd-Warshall/Roy-Floyd Scor 20
Compilator py Status done
Runda Arhiva educationala Marime 0.43 kb
#!/usr/bin/env python3

import sys, itertools

sys.stdout = open('royfloyd.out', 'w')

with open('royfloyd.in', 'r') as fin:
    N = int(fin.readline())
    A = [list(map(int, fin.readline().split())) for _ in range(N)]
    D = [A[i][:] for i in range(N)]

    for k, i, j in itertools.product(range(N), range(N), range(N)):
        D[i][j] = min(D[i][j], D[i][k] + D[k][j])

    for i in range(N):
        print(' '.join(map(str, D[i])))