[Swift] 플로이드 워셜 알고리즘

플로이드 워셜 알고리즘

 

✅ 플로이드 워셜 알고리즘

 

이번 알고리즘 코드는 나동빈 책을 기반으로 공부하고, swift로 제 이해를 바탕으로 직접 코드를 작성하였습니다.

따라서 성능 및 코드 검증이 완벽하지 않아서 오류가 있을 수도 있습니다. 오류가 있다면 댓글로 제보해주세요!

 

 

다익스트라 : 한 지점에서 다른 특정 지점까지 - 그리디에 속함

플로이드 워셜 : 모든 지점에서 다른 모든 지점까지 - dp에 속함

 

 

 

시간복잡도는 O(N^3)이다.

 

 

✅ 코드

import Foundation

/// 정점간의 연결관계, 가중값, 정점의 개수
func FloydWarshall(graph: [[Int]], weight: [Int], n: Int) -> [[Int]] {
    var node = [[Int]](repeating: [Int](repeating: Int(1e9), count: n), count: n)

    for i in 0..<graph.count {
        let x = graph[i][0] - 1
        let y = graph[i][1] - 1
        
        node[x][y] = weight[i]
    }
    
    for i in 0..<n {
        node[i][i] = 0
    }
    
    for i in 0..<n {
        for j in 0..<n {
            for k in 0..<n {
                node[j][k] = min(node[j][i] + node[i][k], node[j][k])
            }
        }
    }
    
    return node
}

struct FloydWarshall_Module {
    
    static let graph = [[1,2], [1,4], [2,1], [2,3], [3,1], [3,4], [4,3]]
    static let weight = [4, 6, 3, 7, 5, 4, 2]
    
    static func run(n: Int) {
        let answer = FloydWarshall(graph: graph, weight: weight, n: 4) // [[0, 4, 8, 6], [3, 0, 7, 9], [5, 9, 0, 4], [7, 11, 2, 0]]
        print(answer)
    }
}