[프로그래머스/python] 깊이/너비 우선 탐색(DFS/BFS) : 네트워크

문제 설명

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

정답

def solution(n, computers):
    answer = 0
    graph = {}
    for i in range(n):
        graph[i] = list(filter(lambda x: computers[i][x] == 1, range(n)))

    visited = []
    for i in range(n):
        if i not in visited:
            # DFS
            stack = [i]
            while stack:
                k = stack.pop()
                if k not in visited:
                    visited.append(k)
                    stack.extend(graph[k])

            answer += 1

    return answer

---

DFS를 활용하기 위해 graph 딕셔너리를 만들어 각각 어떤 컴퓨터와 연결되어 있는지 저장한다.

filter와 람다함수를 사용하여 해당 인덱스의 이차원배열 값이 1인 두번째 인덱스들을 리스트 형태로 딕셔너리에 저장한다.

 

방문했음을 나타낼 visited 리스트를 만들고, 첫번째 인덱스 i가 visited에 없다면 DFS를 실행한다.

DFS를 한번 수행할 때마다 네트워크가 하나 만들어지는 것이므로 answer에 1을 더한다.

 

아래 포스팅은 DFS의 자세한 설명이다.

2021.04.05 - [🔍알고리즘/🔅개념정리] - [파이썬/python] DFS와 BFS

[파이썬/python] DFS와 BFS