프로그래머스 - 네트워크

문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.

컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

제한사항

• 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.
• 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.
• i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computers[i][j]를 1로 표현합니다.
• computer[i][i]는 항상 1입니다.

입출력 예

ncomputersreturn

3	[[1, 1, 0], [1, 1, 0], [0, 0, 1]]	2
3	[[1, 1, 0], [1, 1, 1], [0, 1, 1]]	1

입출력 예 설명

예제 #1
아래와 같이 2개의 네트워크가 있습니다.

예제 #2
아래와 같이 1개의 네트워크가 있습니다.

 

DFS를 이용해서 풀어야 하는 문제

 

visited라는 배열이 각 컴퓨터의 index 순서대로 방문했는지 안했는지 담는 변수

그래서 dfs함수를 반복하면서 계속 연결된 네트워크를 찾아간다.

 

dfs함수에서 item은 computers의 연결에 대한 정보, visited는 방문여부

 

dfs함수에 계속 새로운 컴퓨터 정보를 넣으면서 네트워크에 연결된 컴퓨터를 찾음

이때 visited가 0인 경우에만 찾아야함 1인 경우는 이미 그 컴퓨터와 연결된 노드를 찾았기 때문

func dfs(_ item: [Int], computer: [[Int]], visited: inout [Int]) {
    for i in 0 ... item.count - 1 {
        if item[i] == 1 {
            if visited[i] == 0 {
                visited[i] = 1
                dfs(computer[i], computer: computer, visited: &visited)
            }
        }
    }
}

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