[Python] 백준 1167번 : 트리의 지름

문제

트리의 지름이란, 트리에서 임의의 두 점 사이의 거리 중 가장 긴 것을 말한다. 트리의 지름을 구하는 프로그램을 작성하시오.

입력

트리가 입력으로 주어진다. 먼저 첫 번째 줄에서는 트리의 정점의 개수 V가 주어지고 (2 ≤ V ≤ 100,000)둘째 줄부터 V개의 줄에 걸쳐 간선의 정보가 다음과 같이 주어진다. 정점 번호는 1부터 V까지 매겨져 있다.

먼저 정점 번호가 주어지고, 이어서 연결된 간선의 정보를 의미하는 정수가 두 개씩 주어지는데, 하나는 정점번호, 다른 하나는 그 정점까지의 거리이다. 예를 들어 네 번째 줄의 경우 정점 3은 정점 1과 거리가 2인 간선으로 연결되어 있고, 정점 4와는 거리가 3인 간선으로 연결되어 있는 것을 보여준다. 각 줄의 마지막에는 -1이 입력으로 주어진다. 주어지는 거리는 모두 10,000 이하의 자연수이다.

출력

첫째 줄에 트리의 지름을 출력한다.

예제 입력 1

5
1 3 2 -1
2 4 4 -1
3 1 2 4 3 -1
4 2 4 3 3 5 6 -1
5 4 6 -1

예제 출력 1

11

 

소스코드

from collections import deque
import sys

read = sys.stdin.readline
v = int(read())
graph = [[] for _ in range(v+1)]


for i in range(v):	// graph[i] = [i번째 노드와 연결된 노드, i번째 노드와 연결된 노드까지의 거리]
    li = list(map(int, read().split()))
    for j in range(1, len(li)-2, 2):
        graph[li[0]].append((li[j], li[j+1]))

def bfs(x):
    visit = [-1] * (v+1)
    max_val = [0, 0]	// [가장 먼 노드까지의 거리, 가장 먼 노드의 번호]
    visit[x] = 0
    queue = deque([x])

    while queue:
        f = queue.popleft()
        for a, b in graph[f]:
            if visit[a] == -1:
                visit[a] = visit[f] + b
                queue.append(a)
                if max_val[0] < visit[a]:
                    max_val = visit[a], a

    return max_val

val, node = bfs(1)

val, node = bfs(node)
print(val)

1. 임의의 한 노드에서 출발하여 bfs를 이용해 그 노드에서 가장 먼 노드를 구합니다. -> 가장 먼 노드의 번호를 구할 수 있습니다.

2. 1번에서 구한 노드에서 bfs를 한 번 더 진행하여 다시 가장 먼 노드를 구합니다. -> 가장 먼 노드까지의 거리를 구할 수 있습니다.

 

 

출처 : https://velog.io/@coding_egg/%EB%B0%B1%EC%A4%80-1991%EB%B2%88-%ED%8A%B8%EB%A6%AC%EC%9D%98-%EC%A7%80%EB%A6%84-python-%ED%8C%8C%EC%9D%B4%EC%8D%AC

[백준] 1167번 : 트리의 지름 (python 파이썬)