[프로그래머스] 가장 먼 노드

 

 

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

 

 

 

 

from collections import deque

def solution(n, edge):
    graph = [[] for i in range(n + 1)]
    node_count = [0 for i in range(n + 1)]
    for i, e in enumerate(edge):
        graph[e[0]].append(e[1])
        graph[e[1]].append(e[0])
    
    node_count[1] = 1
    queue = deque()
    queue.append(1)
    
    while queue:
        now = queue.popleft()
        for g in graph[now]:
            if node_count[g] == 0:
                queue.append(g)
                node_count[g] = node_count[now] + 1
    
    max_count = max(node_count)
    return node_count.count(max_count)