結果

問題 No.1098 LCAs
コンテスト
ユーザー norioc
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

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