結果

問題 No.1418 Sum of Sum of Subtree Size
コンテスト
ユーザー ntuda
提出日時 2025-06-05 22:48:20
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 336 ms / 2,000 ms
コード長 498 bytes
コンパイル時間 460 ms
コンパイル使用メモリ 82,644 KB
実行使用メモリ 182,272 KB
最終ジャッジ日時 2025-06-05 22:48:31
合計ジャッジ時間 9,749 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(100050)
N = int(input())
AB = [list(map(int,input().split())) for _ in range(N-1)]
E = [[] for _ in range(N)]
for a, b in AB:
    a -= 1
    b -= 1
    E[a].append(b)
    E[b].append(a)
size = [0] * N
ans = 0
def dfs(x,p = -1):
    global ans
    ret = 0
    for y in E[x]:
        if y != p:
            tmp = dfs(y,x)
            ans += (N - tmp) * tmp
            ret += tmp
    ans += (ret + 1) * (N - ret - 1)
    ans += N
    return ret + 1

dfs(0)
print(ans)
0