library_checker/math/
bitwise_and_convolution.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{
4    algebra::AddMulOperation,
5    math::{BitwiseandConvolve, BitwiseorConvolve, ConvolveSteps},
6    num::montgomery::MInt998244353,
7};
8
9#[verify::library_checker("bitwise_and_convolution")]
10pub fn bitwise_and_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, a: [MInt998244353; 1 << n], b: [MInt998244353; 1 << n]);
14    let c = BitwiseandConvolve::<AddMulOperation<_>>::convolve(a, b);
15    iter_print!(writer, @it c);
16}
17
18#[verify::library_checker("bitwise_and_convolution")]
19pub fn bitwise_or_convolution(reader: impl Read, mut writer: impl Write) {
20    let s = read_all_unchecked(reader);
21    let mut scanner = Scanner::new(&s);
22    scan!(scanner, n, mut a: [MInt998244353; 1 << n], mut b: [MInt998244353; 1 << n]);
23    a.reverse();
24    b.reverse();
25    let mut c = BitwiseorConvolve::<AddMulOperation<_>>::convolve(a, b);
26    c.reverse();
27    iter_print!(writer, @it c);
28}