플로이드

[snughnu]

🔗 문제 링크

• 링크: https://www.acmicpc.net/problem/11404
• 난이도: 골드4

✔️ 소요된 시간

50m + 3m

⏰ 시간 복잡도

O(n^3)

[gemini-code-assist]

Code Review

This pull request implements the Floyd-Warshall algorithm for calculating all-pairs shortest paths. The review feedback suggests two primary optimizations: first, populating the distance matrix directly from the input to eliminate the memory overhead of an intermediate adjacency list; and second, aggregating the final output into a single string before printing to improve performance compared to multiple print calls in a loop.

[gemini-code-assist]

graph 인접 리스트를 생성한 뒤 다시 dist 행렬로 옮기는 과정은 메모리와 시간 측면에서 비효율적입니다. 입력을 읽으면서 직접 dist 행렬의 값을 최소값으로 갱신하도록 수정하면 코드가 더 간결해집니다.

// dist 초기화
var dist: [[Int]] = Array(repeating: Array(repeating: INF, count: n+1), count: n+1)
for i in 1...n {
    dist[i][i] = 0
}

// 입력 처리 및 dist 행렬 채우기
for _ in 0..<m {
    let line = readLine()!.split(separator: " ").compactMap { Int($0) }
    let (a, b, c) = (line[0], line[1], line[2])

    dist[a][b] = min(dist[a][b], c)
}

[gemini-code-assist]

반복문 내에서 print를 여러 번 호출하는 대신, map과 joined를 사용하여 전체 결과를 하나의 문자열로 구성한 뒤 한 번에 출력하는 것이 성능 면에서 더 유리합니다.

let result = (1...n).map { i in
    (1...n).map { j in
        dist[i][j] == INF ? "0" : String(dist[i][j])
    }.joined(separator: " ")
}.joined(separator: "\n")
print(result)