프로그래머스 가장 먼 노드 파이썬(PYTHON)

문제링크 : programmers.co.kr/learn/courses/30/lessons/49189

코딩테스트 연습 - 가장 먼 노드

 

from collections import deque
def solution(n, edge):
    graph = [[] for i in range(n+1)]
    for i in edge:
        graph[i[0]].append(i[1])
        graph[i[1]].append(i[0])
    queue = deque()
    visited=set()
    queue.append(1)
    visited.add(1)
    distance=[0 for i in range(n+1)]
    while queue:
        node = queue.popleft()
        for j in graph[node]:
            if j not in visited:
                visited.add(j)
                queue.append(j)
                distance[j] = distance[node]+1
    answer = distance.count(max(distance))
    return answer