結果
問題 |
No.1418 Sum of Sum of Subtree Size
|
ユーザー |
|
提出日時 | 2021-03-11 14:19:26 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 307 ms / 2,000 ms |
コード長 | 597 bytes |
コンパイル時間 | 257 ms |
コンパイル使用メモリ | 82,020 KB |
実行使用メモリ | 112,188 KB |
最終ジャッジ日時 | 2024-10-13 02:47:10 |
合計ジャッジ時間 | 9,030 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
n=int(input()) ab=[list(map(int,input().split())) for _ in range(n-1)] ki=[[] for _ in range(n)] for a,b in ab: a,b=a-1,b-1 ki[a].append(b) ki[b].append(a) # tree_order todo=[[0,-1]] order=[] while todo: v,p=todo.pop() order.append(v) for nv in ki[v]: if nv!=p: todo.append([nv,v]) order.reverse() to_ary=[1]*n seen=[0]*n ans=[0]*n for v in order: seen[v]=1 for nv in ki[v]: if seen[nv]==0: ans[v]+=(n-to_ary[v])*(to_ary[v]) # 根の個数 x サイズ to_ary[nv]+=to_ary[v] else: ans[v]+=(n-to_ary[nv])*(to_ary[nv]) ans[v]+=n print(sum(ans))