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

문제

https://school.programmers.co.kr/learn/courses/30/lessons/43162

프로그래머스

풀이

BFS로 문제를 풀었다.

dq.popleft한 값을 저장하고 그 값에 연결된 노드를 탐색해야하는데.. 계속 삽질했다

코드

from collections import deque


def solution(n, computers):
    answer = 0
    vis = [False]*n
    for i in range(n): #컴퓨터 개수만큼 순회
        if(vis[i] == False): #i번째 방문 안했으면
            answer+=1
            dq = deque()
            dq.append(i) #큐에 넣고 
            vis[i] = True #방문 표시
            while(dq):
                v = dq.popleft() #큐에서 나오면서 정보 저장
                for j in range (n):
                    if(vis[j]==False and computers[v][j]==1): #큐에서 나온애와 연결된 인접노드 탐색
                        dq.append(j)
                        vis[j] = True
    return answer