AcWing 1052. 设计密码

#include<bits/stdc++.h>
using namespace std;

const int N = 55,MOD = 1e9+7;
char str[N];
int ne[N];
int f[N][N];
//对于前i个字符，匹配到str字符串的第j个位置（匹配成功说明包含子串），计算不包含子串的数量
int n,m;

int main()
{
    scanf("%d%s",&n,str+1);
    m=strlen(str+1);
    //ne
    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++)
            {
                int u=j;//下个字符是k时，u来更新j的位置
                while(u&&str[u+1]!=k)u=ne[u];
                if(str[u+1]==k)u++;
                //还未匹配成功
                if(u<m)
                {
                    //k为第i+1个字符且还没包含子串时，将前一种状态的方案数加上
                    f[i+1][u]=(f[i+1][u]+f[i][j])%MOD;
                }
            }
        }
    }
    int res=0; 
    for(int i=0;i<m;i++)
    {
        res=(res+f[n][i])%MOD;
    }
    cout<<res;
    return 0;
}

辰风dalao思路：
如何判断某一种方案是可能的呢？联系KMP的子串匹配方法，就是判断对于固定的 i 和 j 判断当前字符是不是和子串中 j+1 的字符匹配，匹配就j++，不匹配 j 就回跳，如果我们到最后，也就是 i = 26 的时候都没办法使 j 到子串的末尾，那么就意味着我们的密码中没有子串，也就是一种可能。