[PROGRAMMERS] 네트워크

김태민·2022년 2월 2일
0

알고리즘

목록 보기
30/77

mingssssss

1. 문제

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

2. 코드

from collections import deque

def solution(n, computers):
    answer = 0
    # 1. 방문함수 생성
    visited = [False for i in range(n)]
    
    # 2. 시작 노드 설정
    for x in range(n):
        if visited[x] == False:
            bfs(n, computers, x, visited)
            answer += 1
    return answer

# 3. bfs 함수 정의
def bfs(n, computers, x, visited):
    # 4. queue 리스트 생성
    queue = deque()
    queue.append(x)
    while queue:
        x = queue.popleft()
        visited[x] = True
        for y in range(n):
            if x != y and computers[x][y] == 1:
                if visited[y] == False:
                    queue.append(y)
    

3. 리뷰

bfs의 2차원 배열 뼈대를 만든 후에, 네트워크의 수를 체크하면 되는 문제이다. bfs로 연결된 모든 노드를 탐색하고 나면 answer에 1를 더해 총 네트워크의 수를 계산했다.
매번 어렵다고 뒤로 미뤄뒀던 bfs를 친구와 함께 풀게 되었다. 머리속에 어떤 흐름인지 감은 오지만 실전 문제를 풀지 않아 막막한 상태였다. 1시간 정도 고민하고 친구와 함께 다른 풀이를 보며 이해했다. 복잡하지만 꼭 내것으로 만들어야겠다고 생각했다.

profile
어제보다 성장하는 개발자

0개의 댓글