백준 1167번 - 트리의 지름

출처: 1167번: 트리의 지름 (acmicpc.net)

1167번: 트리의 지름

---

트리의 지름 성공분류

시간제한 메모리 제한 제출 정답 맞은 사람 정답 비율

2 초

256 MB

15351

5783

4177

35.826%

문제

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

입력

트리가 입력으로 주어진다. 먼저 첫 번째 줄에서는 트리의 정점의 개수 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

 

---

[ 문제 설명 ]

 

트리의 지름이란, 트리에서 임의의 두 점 사이이 거리 중 가장 긴 것을 말한다.

 

트리의 지름을 찾는 방법은 3가지 차례 후 구할 수 있다.

1. 트리에서 임의의 정점 x를 잡는다

2. 정점 x에서 가장 먼 정점 y를 찾는다

3. 정점 y에서 가장 먼 정점 x를 찾는다

 

그림으로 설명해보겠습니다.

 

그림 1 - 트리의 지름 원리 설명

그림 2 - 트리의 지름 코드 설명

 

 

---

graph = dict()
v = int(input())
for i in range(v):
    graph[i + 1] = {}

for i in range(v):
    inputList = list(map(int, input().split(' ')))
    for j in range(1, len(inputList) - 1, 2):
        graph[inputList[0]][inputList[j]] = inputList[j + 1]


# 첫 dfs
toVisit = [(1, 0)]
visited = [False for i in range(v)]
visited[0] = True
maxLength = 0
maxNode = 1

while toVisit:
    now, length = toVisit.pop()
    maxLength, maxNode = (length, now) if length > maxLength else (maxLength, maxNode)
    for i in graph[now].keys():
        
        if not visited[i - 1]:
            visited[i - 1] = True
            toVisit.append((i, length + graph[now][i]))


# 2st dfs
toVisit = [(maxNode, 0)]
visited = [False for i in range(v)]
visited[maxNode - 1] = True
maxLength = 0

while toVisit:
    now, length = toVisit.pop()
    maxLength = length if length > maxLength else maxLength

    for i in graph[now].keys():
        if not visited[i - 1]:
            visited[i - 1] = True
            toVisit.append((i, length + graph[now][i]))


print(maxLength)

 

---

 

후기

 

처음 볼 때 두 번째 볼 때 다른 느낌..