💻코딩테스트 문제풀이 26

Chapter12. 구현

[문제58] 네트워크 - Level3

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

[코드작성]

1. 탐색을 수행할 준비

def solution(n, computers):
	visited = [0] * n
    answer = 0
    
    for i in range(n):
    	if visited[i] == 0:
        	<DFS 함수가 들어갈 위치>
            answer += 1
            
    return answer

2. DFS 함수 구현

def dfs(k, graph, visited):
	visited[k] = 1
    for i in range(len(graph[k])):
    	if visited[i] == 0 and graph[k][i] and graph[k][i] == 1:
        	dfs(i, graph, visited)

[전체코드]

def dfs(k, graph, visited):
	visited[k] = 1
    for i in range(len(graph[k])):
    	if visited[i] == 0 and graph[k][i] and graph[k][i] == 1:
        	dfs(i, graph, visited)
            
def solution(n, computers):
	visited = [0] * n
    answer = 0
    
    for i in range(n):
    	if visited[i] == 0:
        	dfs(i, graph, visited)
            answer += 1
            
    return answer