矩阵快速幂

用vector：

typedef vector<ll> vec;
typedef vector<vec> mat;
map<ll,ll> mp;
mat mul(mat &A, mat &B) {
    mat C(A.size(), vec(B[0].size()));
    for(int i = 0; i < A.size(); i ++) {
        for(int j = 0; j < B[0].size(); j ++) {
            for(int k = 0; k < B.size(); k ++) {
                C[i][j] = (C[i][j] + A[i][k]*B[k][j]);
            }
            C[i][j] %= mod;
        }
    }
    return C;
}

mat pow(mat A, ll n) {
    mat B(A.size(), vec(A[0].size()));
    for(int i = 0; i < B.size(); i ++) B[i][i] = 1;
    while(n) {
        if(n&1LL) B = mul(B,A);
        A = mul(A, A);
        n >>= 1LL;
    }
    return B;
}

用结构体，速度比vector快。

int len;
struct mat{
    ll m[len][len];
    Nod(){
        memset(m, 0, sizeof(m));
    }
};
mat mul(mat &A, mat &B) {
    mat C;
    for(int i = 0; i < len; i ++) {
        for(int j = 0; j < len; j ++) {
            for(int k = 0; k < len; k ++) {
                C.m[i][j] = (C.m[i][j] + A.m[i][k]*B.m[k][j]) % mod;
            }
        }
    }
    return C;
}

mat pow(mat A, ll n) {
    mat B;
    for(int i = 0; i < len; i ++) B.m[i][i] = 1;
    while(n) {
        if(n&1LL) B = mul(B,A);
        A = mul(A, A);
        n >>= 1LL;
    }
    return B;
}

同一个题目，左边是用结构体算出来的时间，右边是用vector算出来的时间。