acwing 1052 设计密码

题面

题解(DP状态机+KMP)

KMP匹配过程： 对于模板串我们先提前预处里出ne数组，然后在匹配的时候，如果当前位置的模式串和模板串匹配，那么指针向后移动，如果当前位置的模式串和模板串不匹配，我们就要将模板串的指针移动到 ne[j] 的位置继续匹配，当 j 移动到模板串最后的位置(等于模板串的长度)，就说明匹配成功，模板串是模式串的字串

对于这道题，要设计出只包含小写字母长度为N并且不含子串T的模式串S，那么我们可以先处理T串的ne数组，对于S来说，要想满足条件，那么在匹配的过程中，最终是不可能匹配成功的，也就是说最终的指针是不可能指向最后的

我们发现，每次匹配时，指针j在T串中的位置来回变化，那么我们就以T串中每一个位置为一个状态，那么在匹配过程中j指向位置的变化就可以看成是状态的转化，T的长度是m，那么状态就有m+1个，因为T串中包含所有小写字母，状态的移动看的是S串中的字母，那么对于每个状态就有26条边连接其他状态，到此状态机模型就形成了，是m的个状态，每个状态有26条边的状态机

f[i] [j] 表示当前生成的密码到 i 位，在 T 串的第 j 个状态 ，对于生成 S 串，我们只需要在状态机里走N步，并且到达不了m这个状态就是合法的S，现在我们只需统计这样的方案数有多少即可

代码

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 55;
const int mod = 1e9 + 7;

int n, m;
int ne[N];
char p[N];
int f[N][N];  //当前生成的密码已经到第i位,状态是字串中第j个字母的方案数

int main() {

    cin >> n >> p + 1;
    m = strlen(p + 1);

    //求模板串的ne数组
    ne[0] = ne[1] = 0;
    for (int i = 2, j = 0; i <= m; i++) {
        while (j && p[i] != p[j + 1]) j = ne[j];
        if (p[i] == p[j + 1]) j++;
        ne[i] = j;
    }

    f[0][0] = 1;  //S串第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 != p[u + 1])u = ne[u];
                if (p[u + 1] == k) u++;
                if (u < m) {//说明没有出现过子串
                    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 << endl;

    return 0;
}