[ProblemSolving] 프로그래머스 - 네트워크(dfs&bfs) [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입니다.

입출력 예

포인트

• dfs => 해당 분기 끝날때까지 탐색하기 때문 (같은 네트워크 상에 존재할 때까지)
• dfs로 해당 분기 탐색 끝나면, 탐색 횟수를 하나 더 늘린다
• 모든 노드 탐색 끝나면, 누적된 탐색 횟수가 네트워크 개수!!

나의 풀이

기본 dfs 동작 과정과 같다. dfs 동작은
1. 탐색 시작 노드 리스트(스택)에 삽입하고 방문 처리
2. 최상단 노드에 방문하지 않은 인접 노드 있다면, 리스트(스택)에 넣고 방문 처리
방문하지 않은 인접 노드 없다면, 리스트(스택)에서 최상단 노드 꺼내기
3. 과정 2를 수행할 수 없을 때까지 반복

코드

def dfs(computers, v, visited):
    visited[v] = 1 # 2. 방문 처리
    for i in range(len(computers)): # 3. 인접 노드 확인하기
        if computers[v][i] == 1 and visited[i] == 0: # 4. 인접노드 존재하고, 방문 안했다면
            dfs(computers,i, visited) # 탐색
            
def solution(n, computers):
    answer = 0
    visited = [0]*n
    
    for v in range(n):
        # 1. 방문 안했다면 탐색 시작
        if visited[v]==0:
            dfs(computers, v, visited)
            answer += 1
    return answer