코테 연습 - (DFS/BFS) - 네트워크

Doyeon Kim·2022년 1월 3일

코딩테스트 공부

목록 보기
7/171

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


처음에 문제를 보았을 때는 감이 잘 안 잡히다가 예제의 그림들을 보고 단순화하여 생각해보았다

첫번째 컴퓨터를 루트 노드로 두고 bfs 방식을 활용하여 탐색한다.
방문한 노드를 vistied에 저장한다.
탐색이 끝나면, visited를 활용하여 방문하지 않은 노드를 루트 노드로 두고 탐색한다.
모든 노드에 대해 탐색이 끝나면 bfs로 탐색을 시작한 횟수를 리턴한다.

from collections import deque

def solution(n, computers):
    answer = 0
    visited = [0]*n

    for i in range(n):
        if visited[i]:
            continue
        answer += 1
        dq = deque([computers[i]])
        while dq:
            node =dq.popleft()
            for j in range(n):
                if node[j] and not visited[j]:
                    dq.append(computers[j])
                    visited[j] = 1


    return answer

코테를 풀면서 가장 어려운건 한국말인거 같다.......

profile
성장하고 도전하는 개발자. 프로그래밍 좋아하세요?

0개의 댓글