[99클럽] 23일차 TIL

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

프로그래머스

 

그래프를 이용한 탐색 문제입니다.

 

 

그래프를 선언해 주고 해당 노드를 방문할 때마다 거리를 늘려주는 방법을 사용하여 저장하고 max값을 뽑아내줍니다.

 

from collections import deque

def solution(n, edge):
    answer = 0
    
    edge = sorted(edge)
    distance = [0] * (n + 1)
    queue = deque()
    graph = [[] for _ in range(n + 1)]
    
    for i in edge:
        graph[i[0]].append(i[1])
        graph[i[1]].append(i[0])
        
    queue.append(1)
    distance[1] = 1
    
    while queue:
        current = queue.popleft()
        for node in graph[current]:
            if distance[node] == 0:
                queue.append(node)
                distance[node] = distance[current] + 1
    max_distance = max(distance)
    
    for j in distance:
        if j == max_distance:
            answer += 1
    
    return answer