위와 같이 ‘트리의 지름’을 구하는 문제. 다른 점은 트리를 입력받을 때 ‘시작점 도착지1 거리1 도착지2 거리2 -1’과 같은 형태로 입력되는 것 이다. 입력되는 줄을 1부터 길이-1까지 2칸씩 진행하여 {시작점: [[도착지, 거리], [도착지1, 거리1]]}과 같은 형태로 저장하였다. 트리를 이동하는 연산은 위 문제와 동일하게 풀었다. 자세한 풀이는 위 문제의 주석에 설명해 놓았다.
\nimport sys\nfrom collections import deque\ninput = sys.stdin.readline\n\ndef dijkstra(graph, start):\n q = deque([[start, 0]])\n visit = [False for _ in range(len(graph)+1)]\n visit[start] = True\n max_value = 0\n max_node = 0\n\n while q:\n node, weight = q.popleft()\n if max_value < weight:\n max_value = weight\n max_node = node\n\n for next in graph[node]:\n if not visit[next[0]]:\n visit[next[0]] = True\n q.append([next[0], weight + next[1]])\n\n return max_node, max_value\n\nn = int(input())\ngraph = {}\nfor _ in range(n):\n l = list(map(int, input().split()))\n start = l[0]\n for i in range(1, len(l)-1, 2):\n if start not in graph:\n graph[start] = [[l[i], l[i+1]]]\n else:\n graph[start].append([l[i], l[i+1]])\n\nmn, mv = dijkstra(graph, 1)\nprint(dijkstra(graph, mn)[1])