結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー 👑 potato167potato167
提出日時 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)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0