結果
問題 | No.1333 Squared Sum |
ユーザー | Kiri8128 |
提出日時 | 2021-01-08 22:54:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,331 ms / 2,000 ms |
コード長 | 1,580 bytes |
コンパイル時間 | 223 ms |
コンパイル使用メモリ | 81,992 KB |
実行使用メモリ | 197,536 KB |
最終ジャッジ日時 | 2024-04-28 04:29:36 |
合計ジャッジ時間 | 34,652 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 43 ms
53,888 KB |
testcase_01 | AC | 41 ms
53,888 KB |
testcase_02 | AC | 43 ms
54,272 KB |
testcase_03 | AC | 1,241 ms
178,300 KB |
testcase_04 | AC | 1,182 ms
181,352 KB |
testcase_05 | AC | 1,286 ms
181,412 KB |
testcase_06 | AC | 1,269 ms
181,316 KB |
testcase_07 | AC | 1,331 ms
182,056 KB |
testcase_08 | AC | 1,200 ms
178,772 KB |
testcase_09 | AC | 1,201 ms
177,880 KB |
testcase_10 | AC | 1,197 ms
181,348 KB |
testcase_11 | AC | 1,186 ms
178,908 KB |
testcase_12 | AC | 1,240 ms
177,860 KB |
testcase_13 | AC | 747 ms
183,360 KB |
testcase_14 | AC | 1,160 ms
184,732 KB |
testcase_15 | AC | 1,163 ms
182,800 KB |
testcase_16 | AC | 45 ms
53,760 KB |
testcase_17 | AC | 45 ms
53,760 KB |
testcase_18 | AC | 46 ms
54,272 KB |
testcase_19 | AC | 45 ms
54,016 KB |
testcase_20 | AC | 45 ms
53,888 KB |
testcase_21 | AC | 46 ms
54,016 KB |
testcase_22 | AC | 45 ms
54,272 KB |
testcase_23 | AC | 48 ms
54,144 KB |
testcase_24 | AC | 44 ms
53,760 KB |
testcase_25 | AC | 42 ms
54,144 KB |
testcase_26 | AC | 1,160 ms
182,560 KB |
testcase_27 | AC | 1,176 ms
183,096 KB |
testcase_28 | AC | 1,160 ms
182,708 KB |
testcase_29 | AC | 764 ms
182,300 KB |
testcase_30 | AC | 547 ms
119,024 KB |
testcase_31 | AC | 326 ms
98,500 KB |
testcase_32 | AC | 726 ms
139,176 KB |
testcase_33 | AC | 607 ms
126,316 KB |
testcase_34 | AC | 960 ms
159,320 KB |
testcase_35 | AC | 749 ms
137,820 KB |
testcase_36 | AC | 460 ms
113,776 KB |
testcase_37 | AC | 469 ms
114,928 KB |
testcase_38 | AC | 540 ms
119,144 KB |
testcase_39 | AC | 864 ms
146,360 KB |
testcase_40 | AC | 1,116 ms
197,536 KB |
testcase_41 | AC | 1,070 ms
195,768 KB |
testcase_42 | AC | 1,085 ms
195,928 KB |
testcase_43 | AC | 1,083 ms
195,036 KB |
ソースコード
# 全方位木DP import sys input = sys.stdin.readline from collections import deque mod = 10 ** 9 + 7 N = int(input()) X = [[] for i in range(N)] for i in range(N-1): x, y, w = map(int, input().split()) X[x-1].append((y-1, w)) X[y-1].append((x-1, w)) P = [-1] * N Q = deque([0]) R = [] D = [0] * N while Q: i = deque.popleft(Q) R.append(i) for a, w in X[i]: if a != P[i]: P[a] = i D[a] = w X[a].remove((i, w)) deque.append(Q, a) # # a0 += b0 # a1 += b1 + b0 * d # a2 += b0 * d ** 2 + 2 * b1 * d + b2 ##### Settings unit = (0, 0, 0) merge = lambda a, b: ((a[0] + b[0]) % mod, (a[1] + b[1]) % mod, (a[2] + b[2]) % mod) ## ↓ この2つは必ず対称的にする(じゃないとバグる) def adj_bu(a, i, p): a0, a1, a2 = a a0 += 1 d = D[i] return (a0, (a1 + a0 * d) % mod, (a0 * d ** 2 + 2 * a1 * d + a2) % mod) def adj_td(a, i, p): a0, a1, a2 = a a0 += 1 d = D[i] return (a0, (a1 + a0 * d) % mod, (a0 * d ** 2 + 2 * a1 * d + a2) % mod) adj_fin = lambda a, i: a[2] ##### ME = [unit] * N XX = [0] * N TD = [unit] * N for i in R[1:][::-1]: p = P[i] XX[i] = adj_bu(ME[i], i, p) ME[p] = merge(ME[p], XX[i]) XX[R[0]] = adj_fin(ME[R[0]], R[0]) for i in R: ac = TD[i] for j, _ in X[i]: TD[j] = ac ac = merge(ac, XX[j]) ac = unit for j, _ in X[i][::-1]: TD[j] = adj_td(merge(TD[j], ac), j, i) ac = merge(ac, XX[j]) XX[j] = adj_fin(merge(ME[j], TD[j]), j) print(sum(XX) * (mod + 1) // 2 % mod)