글 읽기 - bfs 구현 질문 드립니다.

skatn3 3년 전 0

bfs 부분 중에 어떤 부분을 잘못 작성하였는지 잘 모르겠습니다 ㅠㅠ

조언해주시면 감사하겠습니다.

from collections import deque

N = int(input())
M = int(input())
graph = [[0]* (N+1) for _ in range(N+1)]
visited = [0]*(N+1)
for _ in range(M):
    a,b = map(int,input().split())
    graph[a][b] = 1
    graph[b][a] = 1


def bfs(graph,visited,c):
    count = 0
    queue = deque()
    for i in range(len(graph[c])):
        if graph[c][i]==1 and visited[i]==0:
            queue.append(i)
            visited[i]=1
            count+=1
            
    while queue:
        x=queue.popleft()
        for i in range(len(graph[x])):
            if graph[x][i]== 1 and visited[i] == 0 :
                queue.append(i)
                visited[i] = 1
                count+=1
                
    return count

result=bfs(graph,visited,1)
print(result)




    
    

    

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

bfs 구현 질문 드립니다.

skatn3 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말