Skip to main content

library_checker/linear_algebra/
matrix_product.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{algebra::AddMulOperation, math::Matrix, num::montgomery::MInt998244353};
4
5#[verify::library_checker("matrix_product")]
6pub fn matrix_product(reader: impl Read, mut writer: impl Write) {
7    let s = read_all_unchecked(reader);
8    let mut scanner = Scanner::new(&s);
9    scan!(scanner, n, m, k, a: [[MInt998244353; m]; n], b: [[MInt998244353; k]; m]);
10    let a = Matrix::<AddMulOperation<_>>::from_vec(a);
11    let b = Matrix::<AddMulOperation<_>>::from_vec(b);
12    let c = a * b;
13    iter_print!(writer, @it2d c.data);
14}
15
16#[verify::library_checker("matrix_product")]
17pub fn matrix_product_strassen(reader: impl Read, mut writer: impl Write) {
18    let s = read_all_unchecked(reader);
19    let mut scanner = Scanner::new(&s);
20    scan!(scanner, n, m, k, a: [[MInt998244353; m]; n], b: [[MInt998244353; k]; m]);
21    let a = Matrix::<AddMulOperation<_>>::from_vec(a);
22    let b = Matrix::<AddMulOperation<_>>::from_vec(b);
23    let c = a.mul_strassen(&b);
24    iter_print!(writer, @it2d c.data);
25}