bzoj4037 [HAOI2015]数字串拆分（矩阵快速幂+dp）

对于计算f[n],我们显然可以构造出m*m的转移矩阵A来矩阵快速幂求。
然后我们考虑最后的答案，如果是分成了k个数:a1,a2,…,ak。
那么结果就是$A^{a_1+a_2+...+a_k}=A^{a_1}*A^{a_2}*...*A^{a_k}$
最后的答案就是若干个这种矩阵加起来。
满足乘法分配律，因此我们可以进行dp
f[i]表示1~i这些数字的答案矩阵，枚举j来转移，$f[i]+=f[j]*A^{w(j+1,i)}$
w(j+1,i)表示j+1~i的数字构成的数。可以预处理出c[x][i]表示$A^{x*10^i}$来快速转移。复杂度$O(n^2m^3)$

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 510
#define mod 998244353
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m;
char s[N];
inline void inc(int &x,int y){x+=y;x%=mod;}
struct Matrix{
    int a[6][6];
    int* operator[](int x){return a[x];}
    inline void init(bool fl){
        memset(a,0,sizeof(a));if(fl) for(int i=1;i<=m;++i) a[i][i]=1;
    }friend Matrix operator+(Matrix a,Matrix b){
        for(int i=1;i<=m;++i)
            for(int j=1;j<=m;++j) inc(a[i][j],b[i][j]);return a;
    }friend Matrix operator*(Matrix a,Matrix b){
        Matrix res;res.init(0);
        for(int i=1;i<=m;++i)
            for(int j=1;j<=m;++j)
                for(int k=1;k<=m;++k)
                    inc(res[i][j],(ll)a[i][k]*b[k][j]%mod);
        return res;
    }friend Matrix operator^(Matrix x,int k){
        Matrix res;res.init(1);
        for(;k;k>>=1,x=x*x) if(k&1) res=res*x;return res;
    }
}c[10][N],trans,f[N];
int main(){
//  freopen("a.in","r",stdin);
    scanf("%s",s+1);
    m=read();n=strlen(s+1);
    for(int i=1;i<=m;++i){
        trans[i][1]=1;
        if(i+1<=m) trans[i][i+1]=1;
    }for(int i=0;i<=n;++i) c[0][i].init(1);
    for(int i=1;i<=9;++i){
        c[i][0]=c[i-1][0]*trans;
        for(int j=1;j<=n;++j) c[i][j]=c[i][j-1]^10;
    }f[0].init(1);
    for(int i=1;i<=n;++i){
        Matrix tmp;tmp.init(1);
        for(int j=i;j>=1;--j){
            tmp=tmp*c[s[j]-'0'][i-j];
            f[i]=f[i]+f[j-1]*tmp;
        }
    }printf("%d\n",f[n][1][1]);
    return 0;
}