设计密码【状态机模型+KMP】

题意

代码

#include <cstring>
#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
const int N=55,mod=1e9+7;;
int ne[N];//KMP中的next数组
char str[N];//子串
int f[N][N];//表示对于枚举到的那个点的某个状态的解
int  main(){
    int n,m;
    cin >> n >> str+1;
    m=strlen(str+1);
    for(int i=2,j=0;i<=m;i++){
        while(j&&str[i]!=str[j+1])j=ne[j];
        if(str[i]==str[j+1])j++;
        ne[i]=j;
    }
    f[0][0]=1;
    for(int i=0;i<n;i++){//枚举每一个点
        for(int j=0;j<m;j++){//枚举字串中的位置
            for(char k='a';k<='z';k++){//枚举i处的字符
                int u=j;
                while(u&&k!=str[u+1])u=ne[u];
                if(str[u+1]==k)u++;
                if(u<m){//说明没有出现过子串
                    f[i+1][u]=(f[i+1][u]+f[i][j])%mod;//这里最后的状态的f[i+1][u]，所以更新的是这个值
                }
            }
        }
    }
    int res=0;
    for(int i=0;i<m;i++)res = (res + f[n][i]) % mod;//把所有的情况加起来
    cout << res;
}

写的目的：仅仅供自己回忆当时的想法，详细题解点击这里