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

링크: https://programmers.co.kr/learn/courses/30/lessons/43162

코딩테스트 연습 - 네트워크

 

 

dfs 기초 유형입니다.

import Foundation

func solution(_ n:Int, _ computers:[[Int]]) -> Int {
    var result = 0
    var visited = Array(repeating: false, count: n)
    
    func dfs(i: Int) {
        if !visited[i] { // 방문하지 않았다면, i와 연결된 모든 곳을 방문 처리.
            visited[i] = true
            
            for j in 0..<n {
                if computers[i][j] == 1 {
                    dfs(i: j) // i와 이어진 곳이 또 자신과 이어진 곳을 방문 처리.
                }
            }
        }
    }
    
    for i in 0..<n {
        if !visited[i] {
            result += 1
            dfs(i: i)
        }
    }
    
    return result
}

solution(3, [[1, 1, 0], [1, 1, 0], [0, 0, 1]]) // 2
solution(2, [[1, 1, 0], [1, 1, 1], [0, 1, 1]]) // 1