코테

[Programmers] 네트워크 python

29도 맑음 2022. 2. 8. 23:06
반응형

queue문제

맞게 푼거 같은데 15점밖에 안나와서 질문하기 봤더니

단방향해야한다고.... 아니 그림 넘나 양방향처럼 그려놨자나요

#queue쓰고 visited로 하면 될듯?
from collections import deque

def bfs(i, computers, n):
    global visited
    queue = deque()
    queue.append(i)
    while(queue):
        a = queue.popleft()
        for b in range(n):
            if computers[a][b]==1 and not visited[b]:
                visited[b] = True
                queue.append(b)
    return

def solution(n, computers):
    global visited
    answer = 0
    visited = [False]*n
    for i in range(n):
        if not visited[i]:
            visited[i] = True
            bfs(i,computers,n)
            answer+=1
    return answer
반응형