結果
問題 |
No.1418 Sum of Sum of Subtree Size
|
ユーザー |
![]() |
提出日時 | 2023-03-09 07:46:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 449 ms / 2,000 ms |
コード長 | 470 bytes |
コンパイル時間 | 217 ms |
コンパイル使用メモリ | 82,232 KB |
実行使用メモリ | 177,280 KB |
最終ジャッジ日時 | 2024-09-18 02:49:22 |
合計ジャッジ時間 | 8,527 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
import sys sys.setrecursionlimit(10**9) N = int(input()) G = [[] for _ in range(N)] for _ in range(N - 1): a, b = map(int, input().split()) G[a - 1].append(b - 1) G[b - 1].append(a - 1) ans = 0 def dfs(v, pv): global ans cnt = 1 for nv in G[v]: if nv == pv: continue chi = dfs(nv, v) ans += (N - chi) * chi cnt += chi ans += cnt * (N - cnt) ans += N return cnt dfs(0, -1) print(ans)