Algorithm/Programmers
[프로그래머스] 가장 먼 노드 - Python
Dev.sohee
2021. 8. 9. 14:09
https://programmers.co.kr/learn/courses/30/lessons/49189
코딩테스트 연습 - 가장 먼 노드
6 [[3, 6], [4, 3], [3, 2], [1, 3], [1, 2], [2, 4], [5, 2]] 3
programmers.co.kr
from collections import deque
def solution(n, edge):
answer = 0
graph = {i:[] for i in range(1, n+1)}
for i, j in edge:
graph[i].append(j)
graph[j].append(i)
visited = [0] * (n+1)
visited[1] = 1
queue = deque([[1, 0]])
while(queue):
node, depth = queue.popleft()
for i in graph[node]:
if visited[i] == 0:
queue.append([i, depth+1])
visited[i] = depth+1
return visited.count(max(visited))