結果
問題 |
No.1098 LCAs
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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')