[矩阵乘法] BZOJ2326: [HNOI2011]数学作业

我们一个一个接上，应该是 $res=res*10^{t(i)}+i$ 这样的形式，其中$t(i)$ 表示 $i$ 的位数。
怎么样快速算呢，看成递推式：$f(i)=f(i-1)*10^{t(i)}+i$ 。显然可以用矩乘加速。位数相同的一起搞就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
LL n,m,pw[25],MOD;
struct Matrix{
    LL n,m,a[5][5];
    Matrix(){ memset(a,0,sizeof(a)); n=m=0; }
    Matrix operator * (const Matrix &b){
        Matrix c; c.n=n; c.m=b.m;
        for(int i=0;i<c.n;i++)
         for(int j=0;j<c.m;j++) 
          for(int k=0;k<m;k++) (c.a[i][j]+=a[i][k]*b.a[k][j]%MOD)%=MOD;
        return c;
    }
} MT,Ans;
Matrix Pow(Matrix a,LL b){
    Matrix res; res.n=res.m=3; res.a[0][0]=res.a[1][1]=res.a[2][2]=1;
    for(;b;b>>=1,a=a*a) if(b&1) res=res*a;
    return res;
}
int main(){
    pw[0]=1; for(int i=1;i<=18;i++) pw[i]=pw[i-1]*10;
    scanf("%lld%lld",&n,&MOD);
    Ans.n=1; Ans.m=3; Ans.a[0][2]=1;
    for(int i=0;i<=17;i++){
        MT.n=MT.m=3; MT.a[0][0]=pw[i+1]%MOD; MT.a[1][0]=MT.a[1][1]=MT.a[2][0]=MT.a[2][1]=MT.a[2][2]=1;
        LL t=min(n,pw[i+1]-1);
        Ans=Ans*Pow(MT,t-pw[i]+1);
        if(t==n) break;
    }
    if(n==pw[18]) Ans.a[0][0]=(Ans.a[0][0]*(pw[18]%MOD*10%MOD)%MOD+pw[18])%MOD;
    printf("%lld\n",Ans.a[0][0]);
    return 0;
}