結果
問題 |
No.1098 LCAs
|
ユーザー |
![]() |
提出日時 | 2025-04-01 20:40:10 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 621 bytes |
コンパイル時間 | 87 ms |
コンパイル使用メモリ | 12,160 KB |
実行使用メモリ | 59,228 KB |
最終ジャッジ日時 | 2025-04-01 20:40:23 |
合計ジャッジ時間 | 12,337 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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')