library_checker/linear_algebra/
system_of_linear_equations.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{algebra::AddMulOperation, math::Matrix, num::mint_basic::MInt998244353};
4
5#[verify::library_checker("system_of_linear_equations")]
6pub fn system_of_linear_equations(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, a: [[MInt998244353; m]; n], b: [MInt998244353; n]);
10    let a = Matrix::<AddMulOperation<MInt998244353>>::from_vec(a);
11    if let Some(sol) = a.solve_system_of_linear_equations(&b) {
12        iter_print!(writer, sol.basis.len(); @it &sol.particular);
13        for b in sol.basis {
14            iter_print!(writer, @it &b);
15        }
16    } else {
17        iter_print!(writer, -1);
18    }
19}