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

 

코딩테스트 연습 - 네트워크

 

<풀이>

 

BFS를 구현하기 위해 자료구조 Queue를 활용하였다.

 

from collections import deque
 
def solution(n, computers):
    len_c = len(computers)
    visited = [0 for _ in range(len_c)]
    
    answer = 0
    
    for i in range(len_c):
        if not visited[i]:
            answer += 1
            Q = deque([i])
            
            while Q:
                v = Q.popleft()
                
                if not visited[v]:
                    visited[v] = 1
                    for w in range(len_c):
                        if not visited[w] and computers[v][w]:
                            Q.append(w)
    return answer