[swift] 프로그래머스 - 네트워크

프로그래머스

---

나의 풀이

var computer = [[Int]]()
var visited = [Bool]()

func solution(_ n:Int, _ computers:[[Int]]) -> Int {
    computer = computers
    visited = Array(repeating: false, count: n)
    var answer = 0
    for i in 0..<n {
        if !visited[i] {
            dfs(i)
            answer += 1
        }
    }
    return answer
}

func dfs(_ now: Int) {
    if visited[now] { return }
    visited[now] = true
    for i in 0..<computer.count {
        if computer[now][i] == 1 { dfs(i) }
    }
}

풀이 키워드

DFS
깊이 우선 탐색을 재귀함수로 푼 풀이다.
dfs()함수가 반복문을 돌 때 재귀함수의 특성으로
이어진 컴퓨터까지 visited배열에 저장되기 때문에
solution()함수에서 반복문이 돌 때 미리 방문한 computer는 if조건에 해당되지 않는다.
그러므로 총네트워크의 개수를 셀 수 있게 된다.

---