結果

問題 No.1098 LCAs
ユーザー norioc
提出日時 2025-04-01 20:39:44
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 621 bytes
コンパイル時間 354 ms
コンパイル使用メモリ 82,780 KB
実行使用メモリ 139,320 KB
最終ジャッジ日時 2025-04-01 20:39:54
合計ジャッジ時間 9,603 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 20 TLE * 1 -- * 7
権限があれば一括ダウンロードができます

ソースコード

diff #

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

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

    cs = []
    for to in adj[v]:
        if to == par: continue
        cs.append(dfs(to, v))

    for i in range(len(cs)):
        # v と子 i
        ans[v] += cs[i] * 2
        # 子 i と子 j
        for j in range(i+1, len(cs)):
            ans[v] += cs[i] * cs[j] * 2

    return sum(cs) + 1


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