library_checker/math/
lcm_convolution.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{
4    algebra::AddMulOperation,
5    math::{ConvolveSteps, LcmConvolve},
6    num::montgomery::MInt998244353,
7};
8
9#[verify::library_checker("lcm_convolution")]
10pub fn lcm_convolution(reader: impl Read, mut writer: impl Write) {
11    let s = read_all_unchecked(reader);
12    let mut scanner = Scanner::new(&s);
13    scan!(scanner, n, mut a: [MInt998244353; n], mut b: [MInt998244353; n]);
14    a.insert(0, Default::default());
15    b.insert(0, Default::default());
16    let c = LcmConvolve::<AddMulOperation<_>>::convolve(a, b);
17    iter_print!(writer, @it &c[1..]);
18}