[HNOI2008]BZOJ1009 GT考试 - 动态规划 - 矩阵乘法 - KMP

传送门

题解：

首先一开始并没有意识到这个题根KMP有9毛钱关系，只是以为如果设状态为：

f[i][j]表示当前考虑到第i位，且有j位是B的前缀，那么如果下一位是B的j+1位那么就可以转移到f[i+1][j+1]，否则转移到f[i+1][0]。

后来发现我真是个智障。

f[i][j]的j能转移到哪里，计算类似于KMP算法，我们枚举第j+1位是什么转移，转移到哪就在矩阵上+1。

要特判一些情况。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXA 25
#define MAXM 25
using namespace std;
int n,m,mod;
struct matrix{
    int v[MAXA][MAXA],n,m;
    matrix(int _n=0,int _m=0)
    {
        set_size(_n,_m);
    }
    inline void set_size(int _n,int _m)
    {
        n=_n;m=_m;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                v[i][j]=0;
    }
    inline matrix operator=(const matrix &mar)
    {
        n=mar.n,m=mar.m;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                v[i][j]=mar.v[i][j];
        return *this;
    }
    inline matrix operator*(const matrix &b)
    {
        matrix &a=*this,c(a.n,b.m);
        if(a.m!=b.n) return c;
        for(int i=1;i<=a.n;i++)
            for(int k=1;k<=b.m;k++)
                for(int j=1;j<=a.m;j++)
                    c.v[i][k]=(c.v[i][k]+a.v[i][j]*b.v[j][k]%mod)%mod;
        return c;
    }
    friend matrix operator*=(matrix &a,const matrix &b)
    {
        return a=a*b;
    }
}A,B;
int s[MAXM],nxt[MAXM];
inline int getB(int n)
{
    nxt[1]=0;B.v[2][1]=1;B.v[1][1]=9;
    for(int i=2;i<=n;i++)
    {
        for(int j=0;j<=9;j++)
        {
            int t=nxt[i-1];
            while(t&&s[t+1]!=j) t=nxt[t];
            if(s[t+1]==j&&j!=s[i]) B.v[t+2][i]++;
            else B.v[1][i]++;
        }
        int t=nxt[i-1];
        while(t&&s[t+1]!=s[i]) t=nxt[t];
        if(s[t+1]==s[i]) t++;nxt[i]=t;
        B.v[i+1][i]++,B.v[1][i]--;
    }
    return 0;
}
inline matrix fast_pow(const matrix &B,int n)
{
    if(n==1) return B;
    matrix ans=fast_pow(B,n/2);
    ans=ans*ans;
    if(n&1) ans*=B;
    return ans;
}
int main()
{
    scanf("%d%d%d",&n,&m,&mod);
    for(int i=1;i<=m;i++) scanf("%1d",&s[i]);
    A.set_size(m,1);B.set_size(m,m);getB(m);
    A.v[1][1]=9,A.v[2][1]=1;//if n-1==0
    A=fast_pow(B,n-1)*A;int ans=0;
    for(int i=1;i<=m;i++)
        ans=(ans+A.v[i][1])%mod;
    printf("%d\n",ans);return 0;
}