【BZOJ-1009】GT考试 KMP+DP+矩阵乘法+快速幂

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int n,m,p,next[110],an; char S[110];
struct Matrixnode
{
    int da[30][30];
    Matrixnode(){memset(da,0,sizeof(da));}
}a; 
Matrixnode Mul (Matrixnode A,Matrixnode B)
{
    Matrixnode C;
    for (int i=0; i<m; i++)
        for (int j=0; j<m; j++)
            for (int k=0; k<m; k++)
                C.da[i][j]=(C.da[i][j]+A.da[i][k]*B.da[k][j])%p;
    return C;
}
Matrixnode Pow (Matrixnode A,int x)
{
    Matrixnode re;
    for (int i=0; i<m; i++) re.da[i][i]=1;
    for (int i=x; i; i>>=1,A=Mul(A,A))
        if (i&1) re=Mul(re,A);
    return re;
}
void KMP_prework()
{
    for (int j=0,i=2; i<=m; i++)
        {
            while (j && S[i]!=S[j+1]) j=next[j];
            if (S[j+1]==S[i]) j++; next[i]=j;
        }
    for (int i=0; i<m; i++)
        for (int x,j=0; j<10; j++)
            {
                x=i;
                while (x && S[x+1]-‘0‘!=j) x=next[x];
                if (j==S[x+1]-‘0‘) a.da[i][x+1]++; else a.da[i][0]++;
            }
}
int main()
{
    scanf("%d%d%d\n",&n,&m,&p); scanf("%s",S+1);
    KMP_prework();
    Matrixnode ans; ans=Pow(a,n);
    for (int i=0; i<m; i++) an=(an+ans.da[0][i])%p;
    printf("%d\n",an);
    return 0;
}