結果

問題 No.1484 木に数を書き込む問題 / Just Write Numbers! 2
ユーザー lam6er
提出日時 2025-03-20 21:05:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 395 ms / 2,000 ms
コード長 1,039 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 82,644 KB
実行使用メモリ 110,536 KB
最終ジャッジ日時 2025-03-20 21:05:21
合計ジャッジ時間 8,836 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque

def main():
    n = int(sys.stdin.readline())
    adj = [[] for _ in range(n + 1)]
    for _ in range(n - 1):
        a, b = map(int, sys.stdin.readline().split())
        adj[a].append(b)
        adj[b].append(a)
    
    def get_farthest(start):
        visited = [-1] * (n + 1)
        q = deque([start])
        visited[start] = 0
        max_node = start
        max_dist = 0
        while q:
            u = q.popleft()
            for v in adj[u]:
                if visited[v] == -1:
                    visited[v] = visited[u] + 1
                    q.append(v)
                    if visited[v] > max_dist:
                        max_dist = visited[v]
                        max_node = v
        return max_node, max_dist
    
    # First BFS to find one end of the diameter
    u, _ = get_farthest(1)
    # Second BFS to find the other end and compute the diameter length
    v, diameter = get_farthest(u)
    
    print(2 * (n - 1) - diameter)

if __name__ == "__main__":
    main()
0