设计密码【状态机】【KMP】

设计密码

题意

求不包含子串的方案数

思路

匹配子串的过程就是KMP的过程，KMP不跳到m就说明，不包含这个串，

状态表示：f[i] [j] :第i个位置，跳到j的方案数

状态转移： 第i个位置，枚举字符，KMP找到下一个字符以及位置j

代码

#include<bits/stdc++.h>
using namespace std;
const int N=200;
int f[N][N];
char s[N],p[N];
int n,m;
int ne[N];
const int mod=1e9+7;
int main()
{
    cin>>n>>s+1;
    m=strlen(s+1);
    for(int i=2,j=0;i<=m;i++)
    {
        while(j && s[i]!=s[j+1]) j=ne[j];
        if(s[i]==s[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 u='a';u<='z';u++)
            {
                int ne_j=j;
                while(ne_j && u!=s[ne_j+1]) ne_j=ne[ne_j];
                if(u==s[ne_j+1]) ne_j++;
                if(ne_j<m)
                    f[i+1][ne_j] = (f[i+1][ne_j] + f[i][j]) %mod;
            }
        }
    }
    int res=0;
    for(int i=0;i<m;i++) res=(res+f[n][i])%mod;
    cout<<res<<endl;
    return 0;
}