結果
| 問題 | No.2578 Jewelry Store |
| ユーザー |
 suisen
|
| 提出日時 | 2023-01-10 05:23:43 |
| 言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 2,270 bytes |
| コンパイル時間 | 1,536 ms |
| コンパイル使用メモリ | 96,864 KB |
| 実行使用メモリ | 814,176 KB |
| 最終ジャッジ日時 | 2023-12-05 23:30:36 |
| 合計ジャッジ時間 | 3,423 ms |
|
ジャッジサーバーID （参考情報） |
judge14 / judge13 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | WA | - |
| testcase_01 | AC | 2 ms
6,548 KB |
| 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 <map>
#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) {
{
std::vector<int64_t> a2;
std::vector<int32_t> w2;
for (int32_t i = 0; i < n; ++i) if (m % a[i] == 0) {
a2.push_back(a[i]);
w2.push_back(w[i]);
}
a.swap(a2);
w.swap(w2);
n = a.size();
}
// A _ i | m for all i
const int32_t k = pf.size();
std::vector<mint> prd(1 << k, 1);
for (int32_t i = 0; i < n; ++i) {
int32_t t = 0;
for (int32_t j = 0; j < k; ++j) {
if ((m / pf[j]) % a[i] == 0) t |= 1 << j;
}
prd[t] *= 1 + w[i];
}
std::vector<std::pair<int32_t, mint>> prd_vec;
for (int32_t s = 0; s < 1 << k; ++s) {
if (prd[s] != 1) prd_vec.emplace_back(s, prd[s]);
}
mint ans = 0;
for (int32_t s = 0; s < 1 << k; ++s) {
int64_t v = m;
for (int32_t i = 0; i < k; ++i) if ((s >> i) & 1) {
v /= pf[i];
}
mint add = 1;
for (auto [t, pw] : prd_vec) {
if ((s & t) == s) {
add *= pw;
}
}
if ((k - __builtin_popcount(s)) & 1) {
ans -= add;
} else {
ans += add;
}
}
return ans - (m == 1);
}
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;
}