結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー 👑 SPD_9X2SPD_9X2
提出日時 2021-03-05 22:05:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 473 ms / 2,000 ms
コード長 747 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 82,528 KB
実行使用メモリ 220,928 KB
最終ジャッジ日時 2024-10-07 02:01:40
合計ジャッジ時間 7,275 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""

yに関してxがどちら側にいるかだけで変化する
yのサイズをYとする

x == y の時 N
xがyの子の時、(N-Y+1)
xがyの親の時、Y

"""

import sys
from sys import stdin
sys.setrecursionlimit(200000)

ans = 0

def dfs(v,p):

    global ans
    tmp = []
    for nex in lis[v]:
        if nex != p:
            tmp.append ( dfs(nex,v) )

    for x in tmp:
        ans += (N-x) * x

    ans += N
    chnum[v] += sum(tmp)
    rem = N - chnum[v]
    ans += (N-rem) * rem
    
    return chnum[v]
    

N = int(input())

lis = [ [] for i in range(N) ]

for i in range(N-1):

    A,B = map(int,stdin.readline().split())
    A -= 1
    B -= 1

    lis[A].append(B)
    lis[B].append(A)

chnum = [1] * N
dfs(0,0)
print (ans)
0