bzoj1009: [HNOI2008]GT考试

题目
题解
设 dp[i][j] d p [ i ] [ j ] <script type="math/tex" id="MathJax-Element-154">dp[i][j]</script>为第i个号码匹配到第j个不吉利数字的方案数。
显然前i个号码的后缀构成了前j个不吉利数字，而 dp[i][j] d p [ i ] [ j ] <script type="math/tex" id="MathJax-Element-155">dp[i][j]</script>向 i+1 i + 1 <script type="math/tex" id="MathJax-Element-156">i+1</script>转移相当于在后缀后面新加一个字符，可以联想到AC自动机，然而这个只有一个串所以kmp就足够了。
设 a[k][j] a [ k ] [ j ] <script type="math/tex" id="MathJax-Element-157">a[k][j]</script>为k位后面加一个字母转移到j的方案数，于是：
dp[i][j]=Σ0<=k<=m−1dp[i−1][k]∗a[k][j] d p [ i ] [ j ] = Σ 0 <= k <= m − 1 d p [ i − 1 ] [ k ] ∗ a [ k ] [ j ] <script type="math/tex" id="MathJax-Element-158">dp[i][j]=\Sigma_{0<=k<=m−1}dp[i−1][k]∗a[k][j]</script>
我们发现k后面加一个字母转移到j可以用kmp实现。
这个式子是线性的，可以用矩阵优化。

#include<bits/stdc++.h>
using namespace std;
int n,m,K,i,j,k,ne[21],ans;
char s[21];
struct M{
    int a[21][21];
    friend M operator*(M x,M y){
        M z;
        memset(z.a,0,sizeof(z.a));
        for (int i=0;i<m;i++)
            for (int j=0;j<m;j++)
                for (int k=0;k<m;k++) z.a[i][j]=(z.a[i][j]+x.a[i][k]*y.a[k][j])%K;
        return z;
    }
    friend M operator^(M x,int y){
        M z;
        memset(z.a,0,sizeof(z.a));
        for (int i=0;i<m;i++) z.a[i][i]=1;
        for (;y;y>>=1,x=x*x)
            if (y&1) z=z*x;
        return z;
    }
}A,f;
int main(){
    scanf("%d%d%d%s",&n,&m,&K,s);
    for (i=1;i<m;i++){
        while (j && s[i]!=s[j]) j=ne[j-1];
        if (s[i]==s[j]) j++;
        ne[i]=j;
    }
    for (i=0;i<m;i++)
        for (j='0';j<='9';j++){
            k=i;
            while (k && s[k]!=j) k=ne[k-1];
            if (s[k]==j) k++;
            if (k<m) A.a[i][k]++;
        }
    f.a[0][0]=1;
    f=f*(A^n);
    for (i=0;i<m;i++) (ans+=f.a[0][i])%=K;
    printf("%d",ans);
}