結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
👑 ![]() |
提出日時 | 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 |
ソースコード
N=int(input())G=[[]for i in range(N)]for i in range(N-1):a,b=map(int,input().split())a-=1b-=1G[a].append(b)G[b].append(a)order=[0]pare=[-1]*Npare[0]=-2size=[0]*Ndp=[0]*Nfor i in range(N):a=order[i]for x in G[a]:if pare[x]==-1:pare[x]=aorder.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]+=1dp[0]+=size[a]ans=0for 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)