bzoj 1009: [HNOI2008]GT考试 KMP and 矩阵快速幂

题意：给你 1数字串  模数 和长度
问能生成多少个n长度不包含给定数字串的串   0<=a[i]<=9
解：

1、想到dp dp[len][sta] :sta:指的是从前到后已经匹配了多少数组串中的数

N很大直接dp不可能---->矩阵快速幂优化dp

void  kmp(){
    int j=-1,i=0;nxt[0]=-1;int len=strlen(s);
    while(i<len){
        if(j==-1 or s[i]==s[j])
            i+=1,j+=1,nxt[i]=j;
        else j=nxt[j];
    }

    for(int i=0;i<len;i++){
        for(int j=0;j<10;j++){
            char tem='0'+j;
            int t=i;
            while(tem!=s[t] and t!=-1)t=nxt[t];
            t+=2;
            bb[t][i+1]=(bb[t][i+1]+1)%mo;
        }
    }
}

矩阵这样构造：上面的i是现在的sta --> t+1就是下一个sta  之所以t+=2  是为了在矩阵的**下标**从1开始（同理i）。

#include<bits/stdc++.h>
#define mod 51061
#define en '\n'
#define mem(a,b)  memset(a,b,sizeof(a))
#define max(a,b) (a>b? a:b)
#define ll unsigned int
const int maxn = 22;
const int inf=0x3f3f3f3f;
int mo;
using namespace std;
int read() {
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9') {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9') {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int n,m;
char s[maxn];
int bb[maxn][maxn];
int aa[maxn][maxn],nxt[maxn];
void  kmp(){
    int j=-1,i=0;nxt[0]=-1;int len=strlen(s);
    while(i<len){
        if(j==-1 or s[i]==s[j])
            i+=1,j+=1,nxt[i]=j;
        else j=nxt[j];
    }

    for(int i=0;i<len;i++){
        for(int j=0;j<10;j++){
            char tem='0'+j;
            int t=i;
            while(tem!=s[t] and t!=-1)t=nxt[t];
            t+=2;
            bb[t][i+1]=(bb[t][i+1]+1)%mo;
        }
    }
}
void mul(int a[22][22],int b[22][22],int ans[22][22]){
    #define int register int
    int tem[22][22];
    for(int i=1;i<=m;i++){
        for(int j=1;j<=m;j++){
                tem[i][j]=0;
            for(int k=1;k<=m;k++){
                tem[i][j]=(tem[i][j]+a[i][k]*b[k][j])%mo;
            }
        }
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=m;j++){
            ans[i][j]=tem[i][j];
        }
    }
}

signed main() {
#ifdef local
    freopen("input2.txt","r",stdin);
#endif // local
#define int register int
    n=read(),m=read();mo=read();
    scanf("%s",s);
    kmp();
    for(int i=1;i<=m;i++){aa[i][i]=1;}
    while(n){
        if(n&1)mul(aa,bb,aa);
            mul(bb,bb,bb);n>>=1;
    }
    int ans=0;
    for(int i=1;i<=m;i++){
        ans=(ans+aa[i][1])%mo;
    }
    cout<<ans<<en;
    return 0;
}