728x90
from collections import deque
def solution(n, computers):
answer = 0
bfs = deque()
visited = [0]*n
while 0 in visited: # visited 리스트의 모든 값에 방문 표시가 되어있을 때까지 반복
bfs.append(0)
visited[0] = 1
while bfs:
node = bfs.popleft()
for i in range(n):
if visited[i] == 0 and computers[node][i] == 1:
bfs.append(i)
visited[i] = 1
answer += 1 # 한 네트워크의 탐색을 마치면 개수 추가
return answer
728x90
'Algorithm' 카테고리의 다른 글
[Bit Operation] 비트연산 (0) | 2021.04.14 |
---|---|
[프로그래머스] 여행경로 파이썬(DFS) (0) | 2021.03.17 |
[프로그래머스] 단어 변환 파이썬 (BFS/DFS) (0) | 2021.03.17 |
[프로그래머스] 2 x n 타일 파이썬 (DP) (0) | 2021.03.16 |
[프로그래머스] 예상 대진표 파이썬 (DP) (0) | 2021.03.16 |
댓글