結果
問題 |
No.1418 Sum of Sum of Subtree Size
|
ユーザー |
|
提出日時 | 2022-12-13 00:03:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 291 ms / 2,000 ms |
コード長 | 676 bytes |
コンパイル時間 | 177 ms |
コンパイル使用メモリ | 82,584 KB |
実行使用メモリ | 103,500 KB |
最終ジャッジ日時 | 2024-11-06 21:37:23 |
合計ジャッジ時間 | 8,694 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
n = int(input()) g = [[] for _ in range(n)] for _ in range(n - 1): u, v = map(int, input().split()) u -= 1 v -= 1 g[u].append(v) g[v].append(u) dp = [0 for _ in range(n)] par = [None for _ in range(n)] src = 0 st = [~ src, src] while len(st) > 0: cur = st.pop() if cur >= 0: for nxt in g[cur]: if par[cur] == nxt: continue par[nxt] = cur st.append(~ nxt) st.append(nxt) else: dp[~ cur] += 1 if not par[~ cur] is None: dp[par[~ cur]] += dp[~ cur] ans = 0 for y in range(n): ans += n for u in g[y]: if u == par[y]: ans += dp[y] * (n - dp[y]) else: ans += dp[u] * (n - dp[u]) print(ans)