[문제풀이]프로그래머스-네트워크

문제설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 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입니다.

---

해설

문제에서 이야기하는 네트워크는 하나의 연결된 그래프로 생각할 수 있다. 연결정보는 computers라는 2차원 배열형태로 주어진다. 이 computers는 n개의 노드를 가지는 그래프를 인접행렬로 나타낸 것이다.
네트워크의 개수를 세는 것이 목적이므로,
1. 그래프 탐색을 통해 특정 노드와 모든 연결된 노드를 탐색하고, 시작노드를 방문처리한다.
2. computers의 모든 노드에 대해 반복한다.

위 과정을 반복하면서 그래프 탐색을 수행한 횟수를 세면, 네트워크의 개수를 알수 있다. 따라서 그래프 탐색 알고리즘인 DFS를 재귀함수를 통해 구현하고, computers의 모든 노드에 대해 아직 방문하지 않은 경우 DFS를 수행하고, 그 수행 횟수를 기록하면 문제를 풀 수 있다.

---

코드

def dfs(graph, v, visited):
    visited[v] = True
    n = len(graph)
    
    for node in range(n):
        if graph[v][node] == 1 and visited[node] == False:
            dfs(graph, node, visited)
            
def solution(n, computers):
    answer = 0
    visited = [0]*n
    
    for start in range(n): # n개의 node를 각각 시작 노드로 dfs
        if visited[start] == 0: # 이전 탐색에서 방문하지 않았을 경우에만 탐색 시작
            dfs(computers, start, visited)
            answer += 1
    
    return answer

---

🧐 DFS/BFS 정리

문제 풀이에 사용된 DFS알고리즘에 대한 내용은 따로 정리해둔 포스트에서 볼 수 있습니다.

DFS/BFS