結果
問題 | No.1124 Earthquake Safety |
ユーザー | QCFium |
提出日時 | 2020-07-03 14:19:03 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 335 ms / 3,000 ms |
コード長 | 1,657 bytes |
コンパイル時間 | 2,067 ms |
コンパイル使用メモリ | 172,288 KB |
実行使用メモリ | 67,500 KB |
最終ジャッジ日時 | 2023-10-14 23:37:52 |
合計ジャッジ時間 | 15,824 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,372 KB |
testcase_01 | AC | 2 ms
4,368 KB |
testcase_02 | AC | 1 ms
4,368 KB |
testcase_03 | AC | 2 ms
4,372 KB |
testcase_04 | AC | 2 ms
4,372 KB |
testcase_05 | AC | 1 ms
4,372 KB |
testcase_06 | AC | 2 ms
4,376 KB |
testcase_07 | AC | 6 ms
4,368 KB |
testcase_08 | AC | 55 ms
9,072 KB |
testcase_09 | AC | 272 ms
21,024 KB |
testcase_10 | AC | 177 ms
67,500 KB |
testcase_11 | AC | 2 ms
4,372 KB |
testcase_12 | AC | 1 ms
4,372 KB |
testcase_13 | AC | 1 ms
4,372 KB |
testcase_14 | AC | 294 ms
21,040 KB |
testcase_15 | AC | 322 ms
20,812 KB |
testcase_16 | AC | 310 ms
20,956 KB |
testcase_17 | AC | 320 ms
21,140 KB |
testcase_18 | AC | 273 ms
20,992 KB |
testcase_19 | AC | 312 ms
20,952 KB |
testcase_20 | AC | 297 ms
20,876 KB |
testcase_21 | AC | 321 ms
20,976 KB |
testcase_22 | AC | 335 ms
21,020 KB |
testcase_23 | AC | 293 ms
20,776 KB |
testcase_24 | AC | 298 ms
20,772 KB |
testcase_25 | AC | 279 ms
20,792 KB |
testcase_26 | AC | 295 ms
21,096 KB |
testcase_27 | AC | 269 ms
23,332 KB |
testcase_28 | AC | 270 ms
23,316 KB |
testcase_29 | AC | 284 ms
31,928 KB |
testcase_30 | AC | 285 ms
31,760 KB |
testcase_31 | AC | 303 ms
49,520 KB |
testcase_32 | AC | 301 ms
46,152 KB |
testcase_33 | AC | 312 ms
44,720 KB |
testcase_34 | AC | 285 ms
48,736 KB |
testcase_35 | AC | 230 ms
20,544 KB |
testcase_36 | AC | 216 ms
20,612 KB |
testcase_37 | AC | 219 ms
21,828 KB |
testcase_38 | AC | 203 ms
21,612 KB |
testcase_39 | AC | 205 ms
21,356 KB |
testcase_40 | AC | 210 ms
21,872 KB |
testcase_41 | AC | 196 ms
21,572 KB |
testcase_42 | AC | 188 ms
22,132 KB |
testcase_43 | AC | 193 ms
22,300 KB |
testcase_44 | AC | 191 ms
22,064 KB |
testcase_45 | AC | 173 ms
21,724 KB |
testcase_46 | AC | 173 ms
22,420 KB |
testcase_47 | AC | 170 ms
22,000 KB |
testcase_48 | AC | 174 ms
22,352 KB |
testcase_49 | AC | 183 ms
24,892 KB |
testcase_50 | AC | 2 ms
4,368 KB |
testcase_51 | AC | 2 ms
4,376 KB |
testcase_52 | AC | 1 ms
4,372 KB |
testcase_53 | AC | 1 ms
4,376 KB |
testcase_54 | AC | 2 ms
4,368 KB |
testcase_55 | AC | 2 ms
4,368 KB |
testcase_56 | AC | 1 ms
4,372 KB |
testcase_57 | AC | 2 ms
4,372 KB |
ソースコード
#include <bits/stdc++.h> int ri() { int n; scanf("%d", &n); return n; } #define MOD 1000000007 #define inv2 500000004 std::vector<std::vector<int> > hen; std::vector<int> subtree_dsum; void dfs0(int i, int prev) { subtree_dsum[i] = inv2; for (auto j : hen[i]) if (j != prev) { dfs0(j, i); subtree_dsum[i] = (subtree_dsum[i] + (int64_t) subtree_dsum[j] * inv2) % MOD; } } int res = 0; void dfs1(int i, int prev, int prev_dsum) { std::vector<int> all; if (prev != -1) all.push_back(prev_dsum); for (auto j : hen[i]) if (j != prev) all.push_back(subtree_dsum[j]); int sum = std::accumulate(all.begin(), all.end(), 0LL) % MOD; int sum2 = std::accumulate(all.begin(), all.end(), 0, [] (int a, int b) { return (a + (int64_t) b * b) % MOD; }); int sum3 = std::accumulate(all.begin(), all.end(), 0, [] (int a, int b) { return (a + (int64_t) b * b % MOD * b) % MOD; }); res = (res + (int64_t) sum * sum % MOD * sum - (int64_t) sum2 * sum * 3 + sum3 * 2 // choose 3 from different arms + ((int64_t) sum * sum - sum2) * 3 // 1 self + 2 from different arms + (int64_t) sum * 3 + 1) % MOD; // 2 self + 1, 3 self if (res < 0) res += MOD; for (auto j : hen[i]) if (j != prev) { int next = (int64_t) (sum - subtree_dsum[j] + 1) * inv2 % MOD; if (next < 0) next += MOD; dfs1(j, i, next); } } int main() { int n = ri(); hen.resize(n); for (int i = 1; i < n; i++) { int a = ri() - 1; int b = ri() - 1; hen[a].push_back(b); hen[b].push_back(a); } subtree_dsum.resize(n); dfs0(0, -1); dfs1(0, -1, 0); for (int i = 1; i < n; i++) { res += res; if (res >= MOD) res -= MOD; } printf("%d\n", res); return 0; }