[Baekjoon] 1167. 木の直径[G 3]

📚 質問する

https://www.acmicpc.net/problem/1167
前に解いた1967.木の直径の問題とほとんど似ています.
違いは、入力が親と子の関係であり、頂点間の接続ではないことです.
両方の方向に入力があり、すべての入力を受けてアクセス確認を行い、一方の方向でのみ行うことができます.
再帰関数を迂回して、入力したポイントアクセスインデックスを1に変更します.
また、サブノードをチェックするときにゼロのサブノードにアクセスしていない場合は、リーフノードとなります.
このプロシージャは、すべての問題に追加できます.
残りの回答は以下の内容を参照してください.

📌 1967.ツリーの直径ブログの位置付けを参照してください

https://velog.io/@yunhlim/Baekjoon-1967.-%ED%8A%B8%EB%A6%AC%EC%9D%98-%EC%A7%80%EB%A6%84-G4

📒 コード#コード#

def recur(n):
    global total_max
    visited[n] = 1          # 정점 방문 표시
    
    result = []
    for c, w in arr[n]:     # 자식 노드를 재귀호출하면서 가중치 값을 더해준다.
        if not visited[c]:  # 방문했던 노드는 부모 노드로 사용했으니 보지 않는다.
            result.append(w + recur(c))

    if result == []:        # 자식 노드가 없으면 0을 반환
        return 0

    result.sort()           # 노드에 연결된 길이들을 크기 순대로 정렬한다.
    result = result[::-1][0:2]      # 가장 긴 2개만 남긴다.
    total_max = max(total_max, sum(result))  # 2개를 합한 것이 현재 구한 최대 트리의 지름보다 큰지 확인

    return result[0]        # 노드에 연결된 선 중 가장 긴 길이를 리턴


n = int(input())
arr = [[] for _ in range(n + 1)]    # 부모 노드를 인덱스로 하는 리스트(자식노드와 가중치를 담는다.)
visited = [0 for _ in range(n + 1)]
temp = [list(map(int, input().split())) for _ in range(n)]
for i in range(n):
    s = 0
    while temp[i][s] > 0:   # -1이면 종료
        if s == 0:          # 첫 번째 입력을 정점 v1
            v1 = temp[i][s]
            s += 1
        else:
            arr[v1].append([temp[i][s], temp[i][s + 1]])    # 두 번째 입력부터 정점 v2, w로 받아 arr에 넣어준다.
            s += 2
total_max = 0   # 가장 긴 트리의 지름

recur(1)
print(total_max)

🔍 結果