結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー raven7959raven7959
提出日時 2021-09-25 07:21:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 717 ms / 2,000 ms
コード長 769 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,652 KB
実行使用メモリ 273,408 KB
最終ジャッジ日時 2024-07-05 11:58:58
合計ジャッジ時間 9,701 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,736 KB
testcase_01 AC 38 ms
52,352 KB
testcase_02 AC 37 ms
52,096 KB
testcase_03 AC 392 ms
96,640 KB
testcase_04 AC 389 ms
96,948 KB
testcase_05 AC 389 ms
96,980 KB
testcase_06 AC 371 ms
97,408 KB
testcase_07 AC 387 ms
97,152 KB
testcase_08 AC 287 ms
89,960 KB
testcase_09 AC 184 ms
84,096 KB
testcase_10 AC 195 ms
83,776 KB
testcase_11 AC 148 ms
80,512 KB
testcase_12 AC 244 ms
87,884 KB
testcase_13 AC 272 ms
89,728 KB
testcase_14 AC 280 ms
89,552 KB
testcase_15 AC 236 ms
87,392 KB
testcase_16 AC 131 ms
78,592 KB
testcase_17 AC 117 ms
78,516 KB
testcase_18 AC 361 ms
95,616 KB
testcase_19 AC 146 ms
80,336 KB
testcase_20 AC 113 ms
77,852 KB
testcase_21 AC 246 ms
87,296 KB
testcase_22 AC 196 ms
83,968 KB
testcase_23 AC 104 ms
77,112 KB
testcase_24 AC 113 ms
77,952 KB
testcase_25 AC 98 ms
77,392 KB
testcase_26 AC 105 ms
77,804 KB
testcase_27 AC 52 ms
63,360 KB
testcase_28 AC 65 ms
69,888 KB
testcase_29 AC 119 ms
78,368 KB
testcase_30 AC 110 ms
78,164 KB
testcase_31 AC 100 ms
77,440 KB
testcase_32 AC 105 ms
77,696 KB
testcase_33 AC 193 ms
110,720 KB
testcase_34 AC 717 ms
273,408 KB
testcase_35 AC 294 ms
140,368 KB
testcase_36 AC 93 ms
78,608 KB
testcase_37 AC 229 ms
98,876 KB
testcase_38 AC 205 ms
96,500 KB
testcase_39 AC 38 ms
51,712 KB
testcase_40 AC 36 ms
52,352 KB
testcase_41 AC 37 ms
52,608 KB
testcase_42 AC 36 ms
52,736 KB
testcase_43 AC 35 ms
52,480 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**9)
N=int(input())
G=[[] for _ in range(N)]
for _ in range(N-1):
  a,b=map(lambda x:int(x)-1,input().split())
  G[a].append(b)
  G[b].append(a)
subtree_size=[1]*N
subtree_size_of_childlen=[[] for _ in range(N)]
def dfs(G,v,p=-1):
  for nextv in G[v]:
    if nextv!=p:
      dfs(G,nextv,v)
  for c in G[v]:
    if c!=p:
      subtree_size[v]+=subtree_size[c]
def dfs2(G,v,p=-1):
  for nextv in G[v]:
    if nextv!=p:
      dfs2(G,nextv,v)
  for c in G[v]:
    if c!=p:
      subtree_size_of_childlen[v].append(subtree_size[c])
dfs(G,0)
dfs2(G,0)
ans=0
for i in range(N):
  # print(i)
  ans+=(N-subtree_size[i])*subtree_size[i]
  # print(ans)
  for num in subtree_size_of_childlen[i]:
    ans+=(N-num)*num
  # print(ans)
print(ans+N*N)
0