如何构造矩阵快速幂

矩阵快速幂

初步

假如是求F(N)，只要对左边的矩阵乘以N-1次幂，对最终的结果取左上角即是答案

struct Matrix
{
    int m[maxn][maxn]={{0}};
};
Matrix Mul(Matrix A,Matrix B,int n)
{
    Matrix tmp;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            for(int k=1; k<=n; k++)
                tmp.m[i][j]=(tmp.m[i][j]+A.m[i][k]*B.m[k][j])%mod;
    return tmp;
}
Matrix QuickPow(Matrix res,int n,int N)
{
    Matrix ans;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
    {
        if(i==j)ans.m[i][j]=1;
    }
    while(N)
    {
        if(N&1) ans =Mul(ans,res,n);
        res =Mul(res,res,n);
        N = N>>1;
    }
    return ans;
}

变形

已知a[1]=a[2]=a[3]=1，及递推公式 a[i]=a[i-1]+a[i-3]，快速求a[i]
题目链接：https://www.luogu.org/problem/P1939
添加一项 0*a[i-2] 即可，不过构造的矩阵会有变化：

原理还是一样，每次做乘法必须保证 a 矩阵第一列一致性。