matrix

Author: snuke

README.md

@@ -22,6 +22,7 @@ AtCoder解説放送ライブラリ集
 |:--|:--|:--|
 |GCD/LCM|[gcd.cpp](gcd.cpp)|最大公約数と最小公倍数|
 |Combination|[comb.cpp](comb.cpp)|nCkをmod素数で求める|
+|Matrix|[mat.cpp](mat.cpp)|行列|
 
 ### グラフ
 |名前|コード|説明|

mat.cpp

@@ -0,0 +1,31 @@
+template<typename T>
+struct Matrix {
+  int h, w;
+  vector<vector<T>> d;
+  Matrix() {}
+  Matrix(int h, int w, T val=0): h(h), w(w), d(h, vector<T>(w,val)) {}
+  Matrix& unit() {
+    assert(h == w);
+    rep(i,h) d[i][i] = 1;
+    return *this;
+  }
+  const vector<T>& operator[](int i) const { return d[i];}
+  vector<T>& operator[](int i) { return d[i];}
+  Matrix operator*(const Matrix& a) const {
+    assert(w == a.h);
+    Matrix r(h, a.w);
+    rep(i,h)rep(k,w)rep(j,a.w) {
+      r[i][j] += d[i][k]*a[k][j];
+    }
+    return r;
+  }
+  Matrix pow(int t) const {
+    assert(h == w);
+    if (!t) return Matrix(h,h).unit();
+    if (t == 1) return *this;
+    Matrix r = pow(t>>1);
+    r = r*r;
+    if (t&1) r = r*(*this);
+    return r;
+  }
+};