Support dynamic_modint in matrix ops

Author: adamant-pwn

cp-algo/algebra/common.hpp

@@ -43,7 +43,6 @@ namespace cp_algo::algebra {
         }
         return F[n];
     }
-    
     template<typename T>
     T rfact(int n) {
         static std::vector<T> F(maxn);
@@ -57,7 +56,6 @@ namespace cp_algo::algebra {
         }
         return F[n];
     }
-
     template<typename T>
     T small_inv(int n) {
         static std::vector<T> F(maxn);

cp-algo/linalg/vector.hpp

@@ -120,10 +120,9 @@ namespace cp_algo::linalg {
         using Base::Base;
     };
 
-    template<int mod>
-    struct vec<algebra::modint<mod>>:
-            vec_base<vec<algebra::modint<mod>>, algebra::modint<mod>> {
-        using base = algebra::modint<mod>;
+    template<typename base>
+    requires(std::is_base_of_v<algebra::modint_base<base>, base>)
+    struct vec<base>: vec_base<vec<base>, base> {
         using Base = vec_base<vec<base>, base>;
         using Base::Base;
 

verify/algebra/matrix/characteristic.test.cpp renamed to verify/linalg/characteristic.test.cpp

@@ -8,13 +8,15 @@
 using namespace std;
 using namespace cp_algo::linalg;
 using namespace cp_algo::algebra;
+using base = dynamic_modint;
 
 const int mod = 998244353;
 
 void solve() {
+    base::switch_mod(mod);
     int n, m, k;
     cin >> n >> m >> k;
-    matrix<modint<mod>> a(n, m), b(m, k);
+    matrix<base> a(n, m), b(m, k);
     a.read();
     b.read();
     (a * b).print();

verify/algebra/matrix/det.test.cpp renamed to verify/linalg/det.test.cpp

@@ -0,0 +1,33 @@
+// @brief Matrix Product (dynamic modint)
+#define PROBLEM "https://judge.yosupo.jp/problem/matrix_product"
+#pragma GCC optimize("Ofast,unroll-loops")
+#pragma GCC target("avx2,tune=native")
+#include "cp-algo/linalg/matrix.hpp"
+#include <bits/stdc++.h>
+
+using namespace std;
+using namespace cp_algo::linalg;
+using namespace cp_algo::algebra;
+using base = dynamic_modint;
+
+const int mod = 998244353;
+
+void solve() {
+    base::switch_mod(mod);
+    int n, m, k;
+    cin >> n >> m >> k;
+    matrix<base> a(n, m), b(m, k);
+    a.read();
+    b.read();
+    (a * b).print();
+}
+
+signed main() {
+    //freopen("input.txt", "r", stdin);
+    ios::sync_with_stdio(0);
+    cin.tie(0);
+    int t = 1;
+    while(t--) {
+        solve();
+    }
+}