AcWing 1052. 设计密码 DP KMP 状态机

#include <iostream>
#include <cstring>

using namespace std;

const int N = 55, mod = 1e9 + 7;

int n, m;
char s[N];
//f数组其中第一维表示字符串的前缀长度，第二维表示模式串的后缀长度。
//ne数组存储了每个位置的前缀和后缀的最长公共元素长度。
int f[N][N], ne[N];

int main()
{
    cin >> n >> (s + 1);
    m = strlen(s + 1); // 计算模式串T的长度
    //KMP预处理前缀表(得到每个位置的前缀和后缀的最长公共元素长度),next数组
    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;
    }

    //DP

    //在动态规划中
    // f[0][0]表示前缀长度为0,后缀长度为0的字符串的数量
    // 由于空串是任何字符串的子串，因此f[0][0]应该初始化为1
    // 这样，在进行状态转移时，可以保证f[i][j]的值不会因为f[0][0]未被初始化而出错。
    f[0][0] = 1;

    for (int i = 0; i < n; i++) // 遍历密码长度
    {
        for (int j = 0; j < m; j++) // 遍历模式串的长度
        {
            for (char ch = 'a'; ch <= 'z'; ch++)   //枚举第i+1个字符
            {
                int ptr = j;    //计算枚举到第i+1个字符后，后缀匹配的最大长度
                while (ptr && s[ptr + 1] != ch) ptr = ne[ptr]; // 当ptr不为0且s[ptr+1]不等于ch时，将ptr更新为ne[ptr]
                if (s[ptr + 1] == ch) ptr++; // 当s[ptr+1]等于ch时，说明当前字符与前一个字符相同，可以将ptr加1
                f[i + 1][ptr] = (f[i + 1][ptr] + f[i][j]) % mod;    //更新i+1的状态
            }
        }
    }
    //统计所有目标状态
    int res = 0;
    for (int j = 0; j < m; j++) res = (res + f[n][j]) % mod; // 累加所有目标状态的值，并取模
    cout << res << endl;
    return 0;
}