結果
問題 | No.1124 Earthquake Safety |
ユーザー | QCFium |
提出日時 | 2020-07-03 14:53:52 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,114 bytes |
コンパイル時間 | 726 ms |
コンパイル使用メモリ | 10,920 KB |
実行使用メモリ | 205,824 KB |
最終ジャッジ日時 | 2023-10-14 23:40:26 |
合計ジャッジ時間 | 101,872 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 16 ms
8,268 KB |
testcase_01 | AC | 15 ms
8,256 KB |
testcase_02 | AC | 15 ms
8,272 KB |
testcase_03 | AC | 15 ms
8,208 KB |
testcase_04 | AC | 15 ms
8,252 KB |
testcase_05 | AC | 16 ms
8,280 KB |
testcase_06 | AC | 21 ms
8,500 KB |
testcase_07 | AC | 82 ms
10,024 KB |
testcase_08 | AC | 759 ms
27,056 KB |
testcase_09 | AC | 2,407 ms
64,756 KB |
testcase_10 | AC | 2,576 ms
205,824 KB |
testcase_11 | AC | 16 ms
8,260 KB |
testcase_12 | AC | 15 ms
8,256 KB |
testcase_13 | AC | 15 ms
8,256 KB |
testcase_14 | AC | 2,468 ms
64,788 KB |
testcase_15 | AC | 2,465 ms
64,968 KB |
testcase_16 | AC | 2,593 ms
64,908 KB |
testcase_17 | AC | 2,653 ms
64,816 KB |
testcase_18 | AC | 2,553 ms
64,768 KB |
testcase_19 | AC | 2,499 ms
64,908 KB |
testcase_20 | AC | 2,514 ms
64,912 KB |
testcase_21 | AC | 2,479 ms
64,860 KB |
testcase_22 | AC | 2,614 ms
64,860 KB |
testcase_23 | AC | 2,674 ms
65,948 KB |
testcase_24 | AC | 2,635 ms
65,896 KB |
testcase_25 | AC | 2,548 ms
66,684 KB |
testcase_26 | AC | 2,540 ms
66,600 KB |
testcase_27 | AC | 2,566 ms
73,912 KB |
testcase_28 | AC | 2,606 ms
74,052 KB |
testcase_29 | AC | 2,901 ms
98,696 KB |
testcase_30 | AC | 2,857 ms
98,628 KB |
testcase_31 | TLE | - |
testcase_32 | TLE | - |
testcase_33 | TLE | - |
testcase_34 | TLE | - |
testcase_35 | AC | 2,346 ms
61,512 KB |
testcase_36 | AC | 2,281 ms
63,132 KB |
testcase_37 | AC | 2,314 ms
64,768 KB |
testcase_38 | AC | 2,388 ms
63,580 KB |
testcase_39 | AC | 2,369 ms
62,648 KB |
testcase_40 | AC | 2,325 ms
62,812 KB |
testcase_41 | AC | 2,276 ms
62,492 KB |
testcase_42 | AC | 2,223 ms
63,340 KB |
testcase_43 | AC | 2,215 ms
62,552 KB |
testcase_44 | AC | 2,249 ms
62,512 KB |
testcase_45 | AC | 2,320 ms
62,372 KB |
testcase_46 | AC | 2,304 ms
62,724 KB |
testcase_47 | AC | 2,279 ms
62,340 KB |
testcase_48 | AC | 2,230 ms
63,068 KB |
testcase_49 | AC | 2,185 ms
64,780 KB |
testcase_50 | AC | 16 ms
8,260 KB |
testcase_51 | AC | 15 ms
8,220 KB |
testcase_52 | AC | 15 ms
8,220 KB |
testcase_53 | AC | 14 ms
8,184 KB |
testcase_54 | AC | 14 ms
8,272 KB |
testcase_55 | AC | 14 ms
8,200 KB |
testcase_56 | AC | 14 ms
8,256 KB |
testcase_57 | AC | 14 ms
8,200 KB |
ソースコード
import sys sys.setrecursionlimit(400000) MOD = 1000000007 inv2 = 500000004 def dfs0(i, prev) : subtree_dsum[i] = inv2 for j in hen[i] : if j != prev : dfs0(j, i) subtree_dsum[i] = (subtree_dsum[i] + subtree_dsum[j] * inv2) % MOD; res = 0 def dfs1(i, prev, prev_dsum) : global res all = [] if prev != -1 : all.append(prev_dsum) for j in hen[i] : if j != prev : all.append(subtree_dsum[j]) sum1 = sum(all) % MOD sum2 = sum(map(lambda i : i * i, all)) % MOD sum3 = sum(map(lambda i : i * i * i, all)) % MOD res = (res \ + sum1 * sum1 * sum1 - sum2 * sum1 * 3 + sum3 * 2 \ + (sum1 * sum1 - sum2) * 3 \ + sum1 * 3 + 1) % MOD if res < 0 : res += MOD for j in hen[i] : if j != prev : next = (sum1 - subtree_dsum[j] + 1) * inv2 % MOD if next < 0 : next += MOD dfs1(j, i, next); n = int(input()) hen = [[] for i in range(n)] for i in range(n - 1) : a, b = map(int, input().split()) a -= 1 b -= 1 hen[a].append(b) hen[b].append(a) subtree_dsum = [0 for i in range(n)] dfs0(0, -1) dfs1(0, -1, 0) for i in range(n - 1) : res *= 2 if res >= MOD : res -= MOD print(res)