結果

問題 No.1098 LCAs
ユーザー norioc
提出日時 2025-04-01 20:50:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,196 ms / 2,000 ms
コード長 678 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 82,360 KB
実行使用メモリ 301,868 KB
最終ジャッジ日時 2025-04-01 20:50:56
合計ジャッジ時間 12,521 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
import sys
sys.setrecursionlimit(10 ** 6)

try:
    import pypyjit
    pypyjit.set_param('max_unroll_recursion=-1')
except ModuleNotFoundError:
    pass

N = int(input())
adj = defaultdict(list)
for _ in range(N-1):
    u, v = map(lambda x: int(x)-1, input().split())
    adj[u].append(v)
    adj[v].append(u)


def dfs(v, par):
    ans[v] += 1

    nsub = 0  # 部分木の頂点数
    for to in adj[v]:
        if to == par: continue
        r = dfs(to, v)
        # v と子
        ans[v] += r * 2
        # 子同士
        ans[v] += nsub * r * 2
        nsub += r

    return nsub + 1


ans = [0] * N
dfs(0, -1)
print(*ans, sep='\n')
0