矩阵乘法、快速幂

#include <cstdio>
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
// 矩阵的STL实现
typedef vector<int> vec;
typedef vector<vec> mat;
typedef long long ll;
const int MOD = 10000;
// 矩阵乘法
mat mul(mat &A,mat &B){
    mat C(A.size(),vec(B[0].size()));
    for(int i = 0 ; i < A.size() ; i++)
    for(int k = 0 ; k < B.size() ; k++)
    for(int j = 0 ; j < B[0].size() ; j++)
        C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;
    return C;
}
// 矩阵快速幂
mat pow(mat A,ll n){
    mat B(A.size(),vec(A.size()));
    for(int i = 0 ; i < A.size() ; i++)
        B[i][i] = 1;
    while(n > 0){
        if(n & 1) B = mul(B,A);
        A = mul(A,A);
        n >>= 1;
    }
    return B;
}