結果
問題 | No.1333 Squared Sum |
ユーザー | tpyneriver |
提出日時 | 2020-10-31 08:20:49 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 747 ms / 2,000 ms |
コード長 | 1,318 bytes |
コンパイル時間 | 313 ms |
コンパイル使用メモリ | 87,116 KB |
実行使用メモリ | 176,640 KB |
最終ジャッジ日時 | 2023-09-29 10:08:45 |
合計ジャッジ時間 | 20,537 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge11 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 76 ms
71,328 KB |
testcase_01 | AC | 75 ms
71,172 KB |
testcase_02 | AC | 76 ms
71,056 KB |
testcase_03 | AC | 665 ms
159,136 KB |
testcase_04 | AC | 747 ms
158,728 KB |
testcase_05 | AC | 631 ms
161,884 KB |
testcase_06 | AC | 611 ms
159,372 KB |
testcase_07 | AC | 638 ms
161,888 KB |
testcase_08 | AC | 629 ms
162,084 KB |
testcase_09 | AC | 642 ms
159,304 KB |
testcase_10 | AC | 616 ms
159,376 KB |
testcase_11 | AC | 622 ms
146,492 KB |
testcase_12 | AC | 613 ms
159,320 KB |
testcase_13 | AC | 496 ms
159,780 KB |
testcase_14 | AC | 615 ms
161,708 KB |
testcase_15 | AC | 603 ms
161,908 KB |
testcase_16 | AC | 74 ms
71,236 KB |
testcase_17 | AC | 77 ms
71,324 KB |
testcase_18 | AC | 77 ms
71,260 KB |
testcase_19 | AC | 75 ms
71,152 KB |
testcase_20 | AC | 74 ms
71,148 KB |
testcase_21 | AC | 75 ms
71,224 KB |
testcase_22 | AC | 74 ms
71,216 KB |
testcase_23 | AC | 76 ms
70,924 KB |
testcase_24 | AC | 75 ms
71,216 KB |
testcase_25 | AC | 78 ms
70,920 KB |
testcase_26 | AC | 611 ms
158,344 KB |
testcase_27 | AC | 622 ms
158,092 KB |
testcase_28 | AC | 620 ms
158,780 KB |
testcase_29 | AC | 510 ms
158,848 KB |
testcase_30 | AC | 295 ms
112,244 KB |
testcase_31 | AC | 196 ms
96,608 KB |
testcase_32 | AC | 380 ms
127,656 KB |
testcase_33 | AC | 319 ms
114,436 KB |
testcase_34 | AC | 492 ms
127,156 KB |
testcase_35 | AC | 388 ms
127,652 KB |
testcase_36 | AC | 262 ms
108,316 KB |
testcase_37 | AC | 274 ms
108,936 KB |
testcase_38 | AC | 297 ms
112,828 KB |
testcase_39 | AC | 438 ms
135,828 KB |
testcase_40 | AC | 480 ms
171,616 KB |
testcase_41 | AC | 481 ms
173,392 KB |
testcase_42 | AC | 472 ms
173,376 KB |
testcase_43 | AC | 473 ms
176,640 KB |
ソースコード
import sys readline = sys.stdin.readline def parorder(Edge, p): N = len(Edge) par = [0]*N dist = [0]*N par[p] = -1 stack = [p] order = [] visited = set([p]) ast = stack.append apo = order.append while stack: vn = stack.pop() apo(vn) for w, vf in Edge[vn]: if vf in visited: continue visited.add(vf) par[vf] = vn dist[vf] = w ast(vf) return par, order, dist def getcld(p): res = [[] for _ in range(len(p))] for i, v in enumerate(p[1:], 1): res[v].append(i) return res MOD = 10**9+7 N = int(readline()) Edge = [[] for _ in range(N)] for _ in range(N-1): a, b, w = map(int, readline().split()) a -= 1 b -= 1 Edge[a].append((w, b)) Edge[b].append((w, a)) P, L, dist = parorder(Edge, 0) dsum = [0]*N dsum2 = [0]*N size = [1]*N ans = 0 for l in L[:0:-1]: p = P[l] w = dist[l] dsum2[l] = (dsum2[l] + w*w%MOD*size[l] + 2*w*dsum[l])%MOD dsum[l] = (dsum[l] + w*size[l])%MOD ans = (ans + (dsum[l] + dsum[p])**2 - dsum[l]*dsum[l] - dsum[p]*dsum[p] + size[p]*dsum2[l] + size[l]*dsum2[p])%MOD dsum[p] = (dsum[p] + dsum[l])%MOD dsum2[p] = (dsum2[p] + dsum2[l])%MOD size[p] += size[l] print(ans)