11780 플로이드 2

정민용·2023년 4월 13일

백준

목록 보기
132/286

문제

n(1 ≤ n ≤ 100)개의 도시가 있다. 그리고 한 도시에서 출발하여 다른 도시에 도착하는 m(1 ≤ m ≤ 100,000)개의 버스가 있다. 각 버스는 한 번 사용할 때 필요한 비용이 있다.

모든 도시의 쌍 (A, B)에 대해서 도시 A에서 B로 가는데 필요한 비용의 최솟값을 구하는 프로그램을 작성하시오.

# 11780
import sys
input = lambda : sys.stdin.readline().strip()
INF = int(1e9)

n = int(input())
m = int(input())

graph = [[INF] * (n+1) for _ in range(n+1)]
root = [[[i] for i in range(n+1)] for _ in range(n+1)]

for _ in range(m):
    a, b, c = map(int, input().split())
    graph[a][b] = min(graph[a][b], c)
    
for k in range(1, n+1):
    for a in range(1, n+1):
        for b in range(1, n+1):
            if graph[a][k] + graph[k][b] < graph[a][b]:
                graph[a][b] = graph[a][k] + graph[k][b]
                root[a][b] = root[a][k] + root[k][b]
                
for i in range(1, n+1):
    for j in range(1, n+1):
        if graph[i][j] >= INF or i == j:
            graph[i][j] = 0
            root[i][j] = []
            
for g in graph[1:]:
    print(*g[1:])
    
for i in range(1, n+1):
    for j in range(1, n+1):
        if root[i][j]:
            root[i][j] = [i] + root[i][j]
        print(len(root[i][j]), *root[i][j])

백준 11780 플로이드 2

0개의 댓글