矩阵加速线性递推公式

关键词：矩阵加速线性递推公式
用1*2的矩形填充2*n的矩形，共有多少种填充方式？
dp[i]=dp[i-1]+dp[i-2]—————Fibonacci数列
初始值：dp[1]=1,dp[2]=2,定义dp[0]=1,dp[-1]=0;
n很大，需要用矩阵加速运算！
观察下列式子：

其中$f_{-1}$=0,$f_{0}=1$
所以只需要计算矩阵的n次幂即可！
矩阵求幂可以用快速幂加速！

拓展：
1.第二列的两个1改成a和b可以快速求$f_{n}=a*f_{n-2}+b*f_{n-1}$的通项公式
2.同理，可构造k维矩阵快速求出$f_{n}=\sum_{i=n-k}^{n-1}{a_{i}*f_{i}}$的通项公式，矩阵的前k-1列每列仅保留一个1，最后一列为系数列填充$a_{i}$
复杂度：O(k^3*log(n))

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#define mem(a,b) memset(a,sizeof(a),b)
#define mod 19999997
#define ll long long
using namespace std;

const int maxn = 3;

struct Matrax{
    ll m[maxn][maxn];
};
Matrax per,base,ans,xi;//单位矩阵、基矩阵、答案矩阵/系数矩阵
int n;


Matrax multi(Matrax a,Matrax b,int x,int y,int z){//(Ma*Mb)%mod
    Matrax c;
    for(int i=0;i<x;i++)
    for(int j=0;j<z;j++){
        c.m[i][j]=0;
        for(int k=0;k<y;k++){
            c.m[i][j]=(c.m[i][j]+a.m[i][k]*b.m[k][j])%mod;
        }
        c.m[i][j]%=mod;
    }
    return c;
}

Matrax power(Matrax a,int k,int x){//(Ma^k)%mod
    Matrax ans=per,p=a;
    while(k){
        if(k&1){
            ans=multi(ans,p,x,x,x);
            k--;
        }
        else{
            k/=2;
            p=multi(p,p,x,x,x);
        }
    }
    return ans;
}

void init(){//定义单位矩阵和基矩阵
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            if(i==j) per.m[i][j]=1;
            else per.m[i][j]=0;
    for(int i=0;i<2;i++)
    for(int j=0;j<2;j++){
        if(i==0&&j==0) base.m[i][j]=0;
        else base.m[i][j]=1;
    }
    memset(xi.m,0,sizeof(xi.m));
    xi.m[0][0]=0,xi.m[0][1]=1;
}

int main(){
    //freopen("a.txt","r",stdin);
    init();
    while(scanf("%d",&n)!=EOF){
        Matrax tmp=power(base,n,2);
        ans=multi(xi,tmp,1,2,2);
        printf("%lld\n",ans.m[0][1]);
    }
    return 0;
}