結果
| 問題 | 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 |
ソースコード
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')
norioc