Programmers / 3단계 / 가장 먼 노드 / python

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/49189#

 

코드

from collections import deque
def solution(n, edge):
    graph = [[] for _ in range(n+1)]
    visited = [0] * (n+1)
    
    for e in edge:
        x, y = e[0], e[1]
        graph[x].append(y)
        graph[y].append(x)
        
    q = deque()
    q.append(1)
    visited[1] = 1
    while q:
        p = q.popleft()
        for i in graph[p]:
            if not visited[i]:
                visited[i] = visited[p] + 1
                q.append(i)
                
    max_value = max(visited)
    answer = visited.count(max_value)
    
    return answer

 

BFS를 이용하여 문제를 해결할 수 있다.