[백준 1719] 택배

Junyoung Park·2022년 3월 13일
0

코딩테스트

목록 보기
258/631
post-thumbnail

1. 문제 설명

택배

2. 문제 분석

k번째 노드를 일종의 경로로 사용하는 순간 path를 업데이트한다. 플로이드-워셜 알고리즘을 사용했다.

3. 나의 풀이

import sys
import heapq

n, m = map(int, sys.stdin.readline().rstrip().split())
INF = sys.maxsize
nodes = [[INF for _ in range(n+1)] for _ in range(n+1)]
for i in range(1, n+1):
    nodes[i][i] = 0
path = [[0 for _ in range(n+1)] for _ in range(n+1)]

for _ in range(m):
    a, b, c = map(int, sys.stdin.readline().rstrip().split())
    nodes[a][b] = c
    nodes[b][a] = c
    path[a][b] = b
    path[b][a] = a

for k in range(1, n+1):
    for i in range(1, n+1):
        for j in range(1, n+1):
            if nodes[i][j] > nodes[i][k] + nodes[k][j]:
                nodes[i][j] = nodes[i][k] + nodes[k][j]
                path[i][j] = path[i][k]

for i in range(1, n+1):
    path[i][i] = '-'

for i in range(1, n+1):
    print(*path[i][1:], sep=' ')
profile
JUST DO IT

0개의 댓글