結果

問題 No.1098 LCAs
ユーザー anagohirame
提出日時 2020-06-26 21:43:24
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 555 bytes
コンパイル時間 211 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 446,376 KB
最終ジャッジ日時 2024-07-04 20:25:28
合計ジャッジ時間 13,869 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 3 WA * 24 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
sys.setrecursionlimit(10**6)
n = int(input())
adj = [[] for _ in range(n)]
children = [[] for _ in range(n)]
ans = [0 for _ in range(n)]
for _ in range(n-1):
    u, v = map(int, input().split())
    adj[u-1].append(v-1)
    adj[v-1].append(u-1)

def dfs(x, p):
    for v in adj[x]:
        if v == p:
            continue
        dfs(v, x)
    s = sum(children[x])
    res = 2*s+1
    for i in children[x]:
        res += 2 * i * (s-i)
    if x != 0:
        children[p].append(s+1)
    ans[x] = res
    return

dfs(0, -1)
print(*ans, sep="\n")
0