[프로그래머스] 네트워크

기본 아이디어

• 단순한 bfs

코드

from collections import deque

def solution(n, computers):
    cnt = 0
    visited = set()

    for i in range(n):
        if i not in visited:
            cnt += 1
            queue = deque([i])
            visited.add(i)

            while queue:
                node = queue.popleft()
                for j in range(n):
                    if computers[node][j] == 1 and j not in visited:
                        queue.append(j)
                        visited.add(j)

    return cnt

• 엣지를 묻는 문제이다!! 노드가 아님!!!!