矩阵快速幂模板

 

struct Matrix

{

    LL mat[MAXN][MAXN];

    int r;

    void init(int n)

    {

        memset(mat,0,sizeof(mat));

        r=n;

    }

};

Matrix matrix_mul(Matrix a,Matrix b)

{

    Matrix ans;

    ans.init(a.r);

    for(int i=0;i<a.r;i++)

        for(int j=0;j<a.r;j++)

            for(int k=0;k<a.r;k++)

                    ans.mat[i][j]+=a.mat[i][k]*b.mat[k][j];

    return ans;

}

Matrix maxtrix_pow(Matrix a,LL n,int r)

{

    Matrix ans;

    ans.initr(r);

    for(int i=0;i<r;i++) ans.mat[i][i]=1;

    while(n)

    {

        if(n&1) ans=matrix_mul(ans,a);

        n>>=1;

        a=matrix_mul(a,a);

    }

    return ans;

}