프로그래머스(DFS/BFS)-네트워크-Python

문제 설명

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

 

---

내 문제풀이

1. 현재 컴퓨터와 연결된 컴퓨터가 있는 경우, 해당 컴퓨터와 연결된 다른 컴퓨터를 계속해서 탐색하기 위해서 DFS를 사용했다. DFS는 stack, 재귀함수를 활용한다.

2. 네트워크 갯수를 계산하는 answer, 연결된 컴퓨터를 표시하기 위한 stack을 초기화한다.

3. 0번 컴퓨터부터 각 컴퓨터별 네트워크를 확인하기 위해 반복문을 수행한다. 반복문 내에서는 dfs() 함수를 호출한다. 
예외처리로는 2가지 경우를 구현했다.
- 현재 컴퓨터 번호가 stack에 표시되어 있다면, 이미 네트워크에 포함되어 있다는 것이기 때문에 dfs() 를 호출하지 않고 continue를 수행한다. 
- stack의 값이 모두 1이라면, 바로 answer를 반환한다.

4. dfs() 함수는 다음의 총 4개의 인자를 받는다.
n(총 컴퓨터 갯수), idx(현재 컴퓨터 번호), computers(컴퓨터 연결 정보 배열), stack(연결된 컴퓨터 표시)
해당 함수는 현재 컴퓨터와 연결되어 있으며, stack에 표시되지 않은 컴퓨터가 있는 경우를 탐색한다. 이 경우, 새로운 컴퓨터를 stack에 표시하고 해당 컴퓨터 번호로 dfs()를 다시 호출한다. 마지막에는 stack을 반환한다.

5. solution()에서 dfs()를 반복 호출하고, dfs() 내에서 재귀적으로 수행하다보면, 최종 결과 값이 반환된다.

내 코드-파이썬

def dfs(n, idx, computers, stack):
    for i in range(0, n):
        if computers[idx][i]==1 and stack[i]==0: #연결되어 있고, stack에 표시 x
            stack[i] = 1
            dfs(n, i, computers, stack)
    return stack

def solution(n, computers):
    answer = 0 #네트워크 갯수
    stack = [0]*n #연결된 컴퓨터 표시
    for c in range(0, n): #각 컴퓨터별 네트워크 확인
        if stack[c] == 1: continue
        if sum(stack)==n: return answer
        answer += 1        
        stack = dfs(n, c, computers, stack)
        
    return answer