結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー brthyyjpbrthyyjp
提出日時 2021-03-05 21:50:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 182 ms / 2,000 ms
コード長 824 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 81,664 KB
実行使用メモリ 105,600 KB
最終ジャッジ日時 2024-10-07 01:28:52
合計ジャッジ時間 6,261 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import io, os
input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline

n = int(input())
g = [[] for i in range(n)]
for i in range(n-1):
    a, b = map(int, input().split())
    a, b = a-1, b-1
    g[a].append(b)
    g[b].append(a)

s = []
s.append(0)
parent = [-1]*n
order = []
while s:
    v = s.pop()
    order.append(v)
    for u in g[v]:
        if u == parent[v]:
            continue
        s.append(u)
        parent[u] = v
order.reverse()
dp = [1]*n
for v in order:
    if parent[v] != -1:
        dp[parent[v]] += dp[v]
#print(dp)

ans = 0
for v in range(n):
    for u in g[v]:
        if u == parent[v]:
            t = dp[v]*(n-dp[v])
            ans += dp[v]*(n-dp[v])
        else:
            t = (n-dp[u])*dp[u]
            ans += (n-dp[u])*dp[u]
        #print(v, u, t)
ans += n*n
print(ans)
0