[프로그래머스/Python] 네트워크

문제 정보

제목 : 네트워크

난이도 : Lv.3

사용 언어 : Python

문제 링크

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

프로그래머스

 

문제 설명

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

입출력 예

n	computers	return
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개의 네트워크가 있습니다.

---

나의 풀이

컴퓨터 네트워크 연결 정보를 저장할 network defaultdict를 선언하여  디폴트 값으로 list를 지정합니다. n의 길이만큼 반복하는 중첩 for루프를 이용해 네트워크 정보를 저장합니다.

1부터 n+1까지 반복하는 for 루프를 이용해 해당 노드를 기준으로 DFS를 수행합니다. DFS는 현재 노드를 방문했다면 0을 return하고, 방문하지 않았다면 DFS를 수행해 연결된 노드들을 방문했다고 표시 후, 1을 return하여 answer에 더합니다. 반복이 끝나면 answer을 return합니다.

 

코드

from collections import defaultdict

def solution(n, computers):
    answer = 0
    visited = [0 for _ in range(n+1)]
    network = defaultdict(list)
    
    for i in range(n):
        for j in range(n):
            if(computers[i][j] == 1 and i != j) :
                network[i+1].append(j+1)
                
    # 노드별로 DFS 수행
    for i in range(1,n+1):
            answer += DFS(visited, network, i)
    return answer

def DFS(visited,network, node):
    if(visited[node] == 1):
        return 0
    visited[node] = 1
    stack = [node]
    
    while(stack):
        curNode = stack.pop()
        for nextNode in network[curNode]:
            if(visited[nextNode] == 0) :
                visited[nextNode] = 1
                stack.append(nextNode)
    return 1