[Algorithm] 네트워크 (깊이 우선 탐색(DFS))

🖥 문제

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.
컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

🤔 풀이 방법

생각해보자.

• [[1, 1, 0], [1, 1, 0], [0, 0, 1]]
  • 0번과 0번, 1번이 연결되어있다.
  • 1번과 0번, 1번이 연결되어있다.
  • 2번과 2번이 연결되어있다.
  • 0-1, 2와 같은 형태로 2개의 네트워크가 있다.
• [[1, 1, 0], [1, 1, 1], [0, 1, 1]]
  • 0번과 0번, 1번이 연결되어있다.
  • 1번과 0번, 1번, 2번이 연결되어있다.
  • 2번과 1번, 2번이 연결되어있다.
  • 0-1-2와 같은 형태로 1개의 네트워크가 있다.

깊이 우선 탐색이란 루트 노드에서부터 시작해서 다음 브랜치로 넘어가기 전에 해당 브랜치를 완벽하게 탐색하는 방법이다.

0번 노드부터 탐색을 진행한다. 이때 탐색을 진행하는 노드는 전에 방문된 적이 없는 노드여야 한다.

1. 탐색을 시작하면 먼저 방문 체크를 한다.
2. 해당 노드와 연결된 노드가 있고 그 노드가 전에 방문된 적이 없다면 연결된 노드를 탐색한다.

위 1, 2의 과정을 반복하다 더 이상 연결된 노드가 없으면 리턴한다.
연결된 하나의 네트워크를 다 탐색했다. 네트워크의 개수인 answer을 +1 해준다.

방문되지 않은 노드부터 다시 탐색을 시작한다.

👩🏻‍💻 Swift Code

import Foundation

func solution(_ n:Int, _ computers:[[Int]]) -> Int {
    var answer = 0
    var visit = Array(repeating: false, count: n)

    for node in 0..<n where !visit[node] {
        dfs(n, computers, node, &visit)
        answer += 1
    }

    return answer
}

func dfs(_ n: Int, _ computers: [[Int]], _ node: Int, _ visit: inout [Bool]) {
    visit[node] = true
    for i in 0..<n where computers[node][i] == 1 {
        let targetNode = computers[node][i]
        if !visit[i] {
            dfs(n, computers, i, &visit)
        }
    }
}

 

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