結果
問題 | No.2578 Jewelry Store |
ユーザー | suisen |
提出日時 | 2023-01-09 22:05:02 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,900 bytes |
コンパイル時間 | 1,090 ms |
コンパイル使用メモリ | 86,540 KB |
実行使用メモリ | 817,560 KB |
最終ジャッジ日時 | 2024-09-27 00:30:14 |
合計ジャッジ時間 | 3,954 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | WA | - |
testcase_01 | WA | - |
testcase_02 | MLE | - |
testcase_03 | -- | - |
testcase_04 | -- | - |
testcase_05 | -- | - |
testcase_06 | -- | - |
testcase_07 | -- | - |
testcase_08 | -- | - |
testcase_09 | -- | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
testcase_30 | -- | - |
testcase_31 | -- | - |
testcase_32 | -- | - |
testcase_33 | -- | - |
testcase_34 | -- | - |
testcase_35 | -- | - |
testcase_36 | -- | - |
testcase_37 | -- | - |
testcase_38 | -- | - |
testcase_39 | -- | - |
testcase_40 | -- | - |
testcase_41 | -- | - |
testcase_42 | -- | - |
testcase_43 | -- | - |
testcase_44 | -- | - |
testcase_45 | -- | - |
testcase_46 | -- | - |
testcase_47 | -- | - |
testcase_48 | -- | - |
testcase_49 | -- | - |
testcase_50 | -- | - |
testcase_51 | -- | - |
testcase_52 | -- | - |
testcase_53 | -- | - |
ソースコード
#include <algorithm> #include <iostream> #include <vector> #include <atcoder/modint> using mint = atcoder::modint998244353; std::vector<int64_t> prime_factors(int64_t m) { std::vector<int64_t> pf; for (int64_t p = 2; p * p <= m; ++p) { if (m % p == 0) { do m /= p; while (m % p == 0); pf.push_back(p); } } if (m != 1) { pf.push_back(m); } return pf; } mint solve(int32_t n, int64_t m, std::vector<int64_t> pf, std::vector<int64_t> a, std::vector<int32_t> w) { const int32_t k = pf.size(); std::vector<mint> zeta(1 << k, 1); for (int32_t i = 0; i < n; ++i) if (m % a[i] == 0) { int32_t t = 0; for (int32_t j = 0; j < k; ++j) { int64_t p = pf[j]; if ((m / p) % a[i] != 0) { t |= 1 << j; } } zeta[t] *= 1 + w[i]; } // Zeta transform for (int32_t block = 1; block < 1 << k; block *= 2) { for (int32_t offset = 0; offset < 1 << k; offset += 2 * block) { for (int32_t i = 0; i < block; ++i) { zeta[offset + block + i] *= zeta[offset + i]; } } } mint ans = 0; for (int32_t s = 0; s < 1 << k; ++s) { if (__builtin_popcount(s) & 1) { ans -= zeta[s]; } else { ans += zeta[s]; } } return ans; } int main() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); int32_t t; int64_t m; std::cin >> t >> m; auto pf = prime_factors(m); for (int32_t testcase = 0; testcase < t; ++testcase) { int32_t n; std::cin >> n; std::vector<int64_t> a(n); for (auto &e : a) std::cin >> e; std::vector<int32_t> w(n); for (auto &e : w) std::cin >> e; std::cout << solve(n, m, pf, a, w).val() << '\n'; } return 0; }