結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー wolgnikwolgnik
提出日時 2021-03-05 22:10:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 182 ms / 2,000 ms
コード長 649 bytes
コンパイル時間 261 ms
コンパイル使用メモリ 81,664 KB
実行使用メモリ 102,884 KB
最終ジャッジ日時 2024-10-07 02:14:04
合計ジャッジ時間 5,472 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
51,968 KB
testcase_01 AC 39 ms
51,968 KB
testcase_02 AC 36 ms
52,224 KB
testcase_03 AC 182 ms
95,508 KB
testcase_04 AC 175 ms
96,896 KB
testcase_05 AC 181 ms
96,572 KB
testcase_06 AC 173 ms
96,640 KB
testcase_07 AC 180 ms
96,736 KB
testcase_08 AC 128 ms
89,984 KB
testcase_09 AC 90 ms
81,284 KB
testcase_10 AC 91 ms
81,712 KB
testcase_11 AC 82 ms
79,292 KB
testcase_12 AC 119 ms
86,420 KB
testcase_13 AC 130 ms
88,320 KB
testcase_14 AC 128 ms
89,216 KB
testcase_15 AC 111 ms
85,664 KB
testcase_16 AC 69 ms
73,984 KB
testcase_17 AC 70 ms
74,496 KB
testcase_18 AC 165 ms
94,208 KB
testcase_19 AC 77 ms
78,720 KB
testcase_20 AC 62 ms
71,552 KB
testcase_21 AC 107 ms
83,912 KB
testcase_22 AC 100 ms
82,816 KB
testcase_23 AC 64 ms
71,168 KB
testcase_24 AC 66 ms
70,784 KB
testcase_25 AC 68 ms
72,320 KB
testcase_26 AC 63 ms
70,528 KB
testcase_27 AC 45 ms
59,520 KB
testcase_28 AC 54 ms
65,920 KB
testcase_29 AC 65 ms
74,240 KB
testcase_30 AC 64 ms
73,344 KB
testcase_31 AC 57 ms
68,736 KB
testcase_32 AC 63 ms
72,192 KB
testcase_33 AC 76 ms
79,800 KB
testcase_34 AC 151 ms
93,184 KB
testcase_35 AC 90 ms
83,584 KB
testcase_36 AC 66 ms
77,696 KB
testcase_37 AC 133 ms
102,884 KB
testcase_38 AC 128 ms
100,236 KB
testcase_39 AC 34 ms
52,352 KB
testcase_40 AC 34 ms
52,096 KB
testcase_41 AC 35 ms
52,352 KB
testcase_42 AC 33 ms
51,968 KB
testcase_43 AC 37 ms
51,456 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
N = int(input())
e = [[] for _ in range(N + 1)]
for _ in range(N - 1):
  u, v = map(int, input().split())
  e[u].append(v)
  e[v].append(u)

s = [1]
vis = [0] * (N + 1)
parent = [0] * (N + 1)
vis[1] = 1
order = []
while len(s):
  x = s.pop()
  order.append(x)
  for y in e[x]:
    if vis[y]: continue
    parent[y] = x
    vis[y] = 1
    s.append(y)

size = [1] * (N + 1)
for y in order[: : -1]:
  x = parent[y]
  size[x] += size[y]

res = N ** 2
for x in range(1, N + 1):
  for y in e[x]:
    if y == parent[x]:
      res += (N - size[x]) * size[x]
      continue
    res += size[y] * (N - size[y])

print(res)
0