結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 22:31:21 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 723 ms / 2,000 ms |
コード長 | 710 bytes |
コンパイル時間 | 237 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 60,660 KB |
最終ジャッジ日時 | 2024-11-21 06:24:23 |
合計ジャッジ時間 | 12,319 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
import syssys.setrecursionlimit(500000)N = int(input())G = [[] for _ in range(N)]for _ in range(N-1):A,B = map(int,input().split())A-=1;B-=1 #0-indexG[A].append(B)G[B].append(A)P = [-1]*Ndef dfs(v,par=-1):P[v] = 1for u in G[v]:if u == par: continueP[v] += dfs(u,v)return P[v]dfs(0)#print(P)#S = sum(P)global ansans = 0def dfs2(v,par,S): #Sは今の合計global ansans += Sfor u in G[v]:if u == par: continuepre_v = P[v]pre_u = P[u]P[v] = N - P[u]P[u] = Nnew_S = S - pre_u + P[v]dfs2(u,v,new_S)P[v] = pre_vP[u] = pre_udfs2(0,-1,sum(P))print(ans)