bzoj 1009: [HNOI2008]GT考试 -- KMP+矩阵

#include<map>
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define inf 1000000007
#define ll long long
#define N 22
#define F(i,r) for(i=0;i<r;i++)
struct zz{int q[N][N];}a,b;
int n,m,p;
zz operator*(zz k,zz l)
{
    int i,j,o;zz z;
    F(i,m) F(j,m) z.q[i][j]=0;
    F(i,m) F(j,m) F(o,m)
        z.q[i][j]=(z.q[i][j]+k.q[i][o]*l.q[o][j])%p;  
    return z;
}
void ksm(int x)
{
    for(int i=0;i<m;i++) b.q[i][i]=1;
    while(x)
    {
        if(x&1) b=b*a;
        a=a*a;x>>=1;
    }
}
char s[N];
int nxt[N],ans;
int main()
{
    scanf("%d%d%d%s",&n,&m,&p,s+1);
    for(int i=2,j=0;i<=m;i++)
    {
        while(j&&s[i]!=s[j+1]) j=nxt[j];
        if(s[j+1]==s[i]) j++;
        nxt[i]=j;
    }
    for(int i=0;i<m;i++)
    {
        for(int j=0,x;j<10;j++)
        {
            x=i;
            while(x&&s[x+1]-‘0‘!=j) x=nxt[x];
            if(j==s[x+1]-‘0‘) a.q[i][x+1]++;
            else a.q[i][0]++;
        }
    }
    ksm(n);
    for(int i=0;i<m;i++) ans+=b.q[0][i];
    printf("%d\n",ans%p);
    return 0;
}