[HNOI2008]GT考试

题目

[HNOI2008]GT考试

题解

嘛，其实是 KMP+矩阵快速幂优化DP

---

这道题一开始想简单了qwq，认为不能到下一个的都只能往0位置匹配，居然过样例了，坑，后来才发现要用类似AC自动机的思想。

---

考虑裸DP,用dp（i，j）表示还要走i步，当前已经匹配到j位置的方案数。转移其实很好转移，用to（i，j）表示匹配到i位置时如果再填一个j（0<=j<=9）应该匹配到的位置。类似AC自动机。然后转移方程很显然了。

dp（i，j）=sigma（dp（i-1，to（j，k））） （0<=k<=9）

然后考虑构造矩阵加速，对于每个位置j，将矩阵中（j，to（j，k））++。初始状态是dp（0，j）=1。所以将前面矩阵n次方之后能得到dp（n，j），而答案就是dp（n，0）；

---

其实这道题想出来并不复杂啊qwq，不过发现自己想错之后拖了3天才写完（老张的套题好多啊）。写完之后也A不了，拍了一会儿才发现处理to（i，j）的时候写错了，感觉码力达到一个瓶颈，需要突破。

代码

//QWsin
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxm=20+10;

int n,m,MOD,T[maxm],f[maxm],to[maxm][maxm];

#define rep(i) for(int i=1;i<=n;i++)
typedef long long ll;
struct Matrix{
    ll mx[maxm][maxm];int n;
    inline void clear(int x){n=x;memset(mx,0,sizeof mx);}
    inline void init(int x){n=x;clear(n);rep(i) mx[i][i]=1;}
    Matrix operator * (const Matrix &rhs)const{
        Matrix c;c.clear(20);
        rep(i) rep(j) rep(k) (c.mx[i][j]+=mx[i][k]*rhs.mx[k][j])%=MOD;
        return c;
    }
};

void getfail(int *T,int *f)
{
    int k=0;f[1]=0;
    for(int i=2;i<=m;i++)
    {
        while(k&&T[k+1]!=T[i]) k=f[k];
        if(T[k+1]==T[i]) k++;
        f[i]=k;
    }
    for(int i=0;i<=9;i++) to[0][i]=0;
    to[0][T[1]]=1;
    for(int i=1;i<m;i++)
        for(int j=0;j<=9;j++)
        {
            if(j==T[i+1]) to[i][j]=i+1;
            else 
            {
                int k=f[i];
                while(k&&T[k+1]!=j) k=f[k];
                if(T[k+1]==j) to[i][j]=k+1;
                else to[i][j]=0;
            }
        }
}

Matrix Pow(Matrix A,int p)
{
    Matrix ret;ret.init(m);
    for(;p;p>>=1,A=A*A)if(p&1)ret=ret*A;
    return ret;
}

int main()
{
    cin>>n>>m>>MOD;
    for(int i=1;i<=m;i++) scanf("%01d",T+i);
    getfail(T,f);
    Matrix A;A.clear(m);
    for(int i=0;i<m;i++)
        for(int j=0;j<=9;j++)
            A.mx[i+1][to[i][j]+1]++;

    A=Pow(A,n);
    int ans=0;
    for(int i=1;i<=m;i++)
        (ans+=A.mx[1][i])%=MOD;
    cout<<ans;
    return 0;
}