test/cpv/library-checker-convolution/convolution_mod.mints-ms64.conv_anymod-3nttu64.cpp
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | #define AUTO_GENERATED | ||
| 2 | // competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/convolution_mod | ||
| 3 | #include "../../../src/io/container/lib.hpp" | ||
| 4 | #include "../../../src/io/fastin/lib.hpp" | ||
| 5 | #include "../../../src/io/fastout/lib.hpp" | ||
| 6 | |||
| 7 | using namespace tifa_libs; | ||
| 8 | CEXP u32 MOD = 998244353; | ||
| 9 | |||
| 10 | #include "../../../src/conv/add/ntt3/lib.hpp" | ||
| 11 | #include "../../../src/math/ds/mint/ms64/lib.hpp" | ||
| 12 | |||
| 13 | using namespace tifa_libs; | ||
| 14 | using mint = mint_ms64<MOD>; | ||
| 15 | using namespace tifa_libs; | ||
| 16 | using mint_3nttu640 = mint_ms64<167772161>; | ||
| 17 | using mint_3nttu641 = mint_ms64<469762049>; | ||
| 18 | using mint_3nttu642 = mint_ms64<754974721>; | ||
| 19 | using ntt_3nttu640_t = ntt<mint_3nttu640>; | ||
| 20 | using ntt_3nttu641_t = ntt<mint_3nttu641>; | ||
| 21 | using ntt_3nttu642_t = ntt<mint_3nttu642>; | ||
| 22 | using vec_t = vecuu; | ||
| 23 | 53 | auto conv_func(vec_t CR a, vec_t CR b) { | |
| 24 | 53 | ntt_3nttu640_t ntt0; | |
| 25 | 53 | ntt_3nttu641_t ntt1; | |
| 26 | 53 | ntt_3nttu642_t ntt2; | |
| 27 | 53 | return conv_3ntt_u64(ntt0, ntt1, ntt2, a, b, MOD); | |
| 28 | } | ||
| 29 | |||
| 30 | 53 | int main() { | |
| 31 | u32 n, m; | ||
| 32 | 53 | tifa_libs::fin_uint >> n >> m; | |
| 33 |
2/4✓ Branch 1 taken 53 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 53 times.
✗ Branch 5 not taken.
|
159 | vec_t a(n), b(m); |
| 34 | 53 | tifa_libs::fin_uint >> a >> b; | |
| 35 | 53 | tifa_libs::fout << conv_func(a, b) << '\n'; | |
| 36 | 53 | return 0; | |
| 37 | 53 | } | |
| 38 |