[백준] 11404 : 플로이드 - Python

https://www.acmicpc.net/problem/11404

11404번: 플로이드

 

import sys
# Inf의 범위가 작으면 틀릴 수 있으니 주의!!
Inf = 100000000

n = int(sys.stdin.readline())
m = int(sys.stdin.readline())

d = [[Inf] * n for _ in range(n)]

for i in range(m):
    a, b, c = map(int, sys.stdin.readline().split())
    if d[a-1][b-1] > c:
        d[a-1][b-1] = c

for k in range(n):
    for i in range(n):
        for j in range(n):
            # 자기 자신으로 오는 경우
            if j == i:
                d[i][j] = 0
                
            # k를 거쳐서 가는 것, 원래 값 중 최소 값
            else:
                d[i][j] = min(d[i][j], d[i][k] + d[k][j])
                    
for i in range(n):
    for j in range(n):
        if d[i][j] == Inf:
            print(0, end = ' ')
        else:
            print(d[i][j], end = ' ')
    print()