JZYZOJ 1542 [haoi2015]str 矩阵乘法 dp

http://172.20.6.3/Problem_Show.asp?id=1542

dp+矩阵乘法思路hin好想，对于我这种题目稍微学术就几乎什么也不会的人来说唯一的难点在于读题，因为一心想着划水题目没有看清楚，样例wa了一会最后仔细读题发现自己g的操作看错了，非常智障了

代码

 

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm> 
#include<cmath>
using namespace std;
const int maxn=510;
const long long modn=998244353;
char ch[maxn];
int m,n;
struct mat{
    long long e[7][7];
    mat(){memset(e,0,sizeof(e));}
    void cle(){
        for(int i=1;i<=m;i++)e[i][i]=1;
    }
};mat ten[20],f[maxn][maxn],g[maxn];
mat mul(mat x,mat y){
    mat z;
    for(int i=1;i<=m;i++){
        for(int j=1;j<=m;j++){
            for(int k=1;k<=m;k++){
                z.e[i][j]+=x.e[i][k]*y.e[k][j];
                z.e[i][j]%=modn;
            }
        }
    }
    return z;
}
mat plu(mat x,mat y){
    mat z;
    for(int i=1;i<=m;i++){
        for(int j=1;j<=m;j++){
            z.e[i][j]=x.e[i][j]+y.e[i][j];
            z.e[i][j]%=modn;
        }
    }
    return z;
}
mat pow(mat x,long long k){
    mat z;z.cle();
    while(k){
        if(k&1)z=mul(x,z);
        k/=2;x=mul(x,x);
    }
    return z;
}
int main(){
    scanf("%s%d",&ch,&m);n=strlen(ch);
    ten[0].cle();
    for(int i=1;i<m;i++)ten[1].e[i][i+1]=1;
    for(int i=1;i<=m;i++)ten[1].e[m][i]=1;
    for(int i=2;i<=9;i++)ten[i]=pow(ten[1],i);
    for(int i=n;i>=1;i--){
        f[i][i-1].cle();
        for(int j=i;j<=n;j++){
            int z=ch[j-1]-'0';
            f[i][j]=mul(pow(f[i][j-1],10),ten[z]);
        }
    }g[n+1].cle();
    for(int i=n;i>=1;i--){
        for(int j=i;j<=n;j++){
            g[i]=plu(g[i],mul(g[j+1],f[i][j]));
        }
    }
    printf("%I64d
",g[1].e[m][m]);
    return 0;
}