test/cpv/library-checker-convolution/convolution_mod_1000000007.mints-ms64.mints-bs.conv_anymod-3ntt.cpp
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | #define AUTO_GENERATED | ||
| 2 | // competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/convolution_mod_1000000007 | ||
| 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 = 1000000007; | ||
| 9 | |||
| 10 | #include "../../../src/conv/add/ntt3/lib.hpp" | ||
| 11 | #include "../../../src/math/ds/mint/bs/lib.hpp" | ||
| 12 | #include "../../../src/math/ds/mint/ms64/lib.hpp" | ||
| 13 | |||
| 14 | using namespace tifa_libs; | ||
| 15 | using mint = mint_ms64<MOD>; | ||
| 16 | using namespace tifa_libs; | ||
| 17 | using mint_3ntt0 = mint_bs<167772161>; | ||
| 18 | using mint_3ntt1 = mint_bs<469762049>; | ||
| 19 | using mint_3ntt2 = mint_bs<754974721>; | ||
| 20 | using ntt_3ntt0_t = ntt<mint_3ntt0>; | ||
| 21 | using ntt_3ntt1_t = ntt<mint_3ntt1>; | ||
| 22 | using ntt_3ntt2_t = ntt<mint_3ntt2>; | ||
| 23 | using vec_t = vec<mint>; | ||
| 24 | 48 | auto conv_func(vec_t CR a, vec_t CR b) { | |
| 25 | 48 | ntt_3ntt0_t ntt0; | |
| 26 | 48 | ntt_3ntt1_t ntt1; | |
| 27 | 48 | ntt_3ntt2_t ntt2; | |
| 28 | 48 | return conv_3ntt(ntt0, ntt1, ntt2, a, b); | |
| 29 | } | ||
| 30 | |||
| 31 | 48 | int main() { | |
| 32 | u32 n, m; | ||
| 33 | 48 | tifa_libs::fin_uint >> n >> m; | |
| 34 |
2/4✓ Branch 1 taken 48 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 48 times.
✗ Branch 5 not taken.
|
144 | vec_t a(n), b(m); |
| 35 | 48 | tifa_libs::fin_uint >> a >> b; | |
| 36 | 48 | tifa_libs::fout << conv_func(a, b) << '\n'; | |
| 37 | 48 | return 0; | |
| 38 | 48 | } | |
| 39 |