結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー 👑 potato167
提出日時 2022-02-17 01:23:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 257 ms / 2,000 ms
コード長 520 bytes
コンパイル時間 128 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 98,124 KB
最終ジャッジ日時 2024-06-29 07:24:35
合計ジャッジ時間 7,300 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

N=int(input())
G=[[]for i in range(N)]
for i in range(N-1):
	a,b=map(int,input().split())
	a-=1
	b-=1
	G[a].append(b)
	G[b].append(a)
order=[0]
pare=[-1]*N
pare[0]=-2
size=[0]*N
dp=[0]*N
for i in range(N):
	a=order[i]
	for x in G[a]:
		if pare[x]==-1:
			pare[x]=a
			order.append(x)
for i in range(N):
	a=order[N-1-i]
	for x in G[a]:
		if x!=pare[a]:
			size[a]+=size[x]
	size[a]+=1
	dp[0]+=size[a]
ans=0
for i in range(N):
	a=order[i]
	for x in G[a]:
		if pare[x]==a:
			dp[x]=dp[a]+N-2*size[x]
	ans+=dp[a]
print(ans)
0