C++---状态机模型---设计密码（每日一道算法2023.4.14）

SRestia

已于 2023-04-15 11:04:42 修改

阅读量505

点赞数 2

分类专栏：算法文章标签：算法 c++ 开发语言动态规划

于 2023-04-14 17:33:43 首次发布

本文链接：https://blog.csdn.net/srestia/article/details/130158029

版权

算法专栏收录该内容

157 篇文章 5 订阅

订阅专栏

注意事项：
本题为"KMP字符串匹配"的扩展题，建议先阅读这篇文章并理解。
状态机的基本定义：状态机-百度百科。

题目：
你现在需要设计一个密码 S，S 需要满足：

S 的长度是 N；
S 只包含小写英文字母；
S 不包含子串 T；

例如：abc 和 abcde 是 abcde 的子串，abd 不是 abcde 的子串。
请问共有多少种不同的密码满足要求？
由于答案会非常大，请输出答案模 10e9+7 的余数。

输入格式
第一行输入整数N，表示密码的长度。
第二行输入字符串T，T中只包含小写字母。

输出格式
输出一个正整数，表示总方案数模 10e9+7 后的结果。

数据范围
1≤N≤50,
1≤|T|≤N，|T|是T的长度。

输入:
2
a

输出：
625

#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 55, mod = 1000000007;
int n, m;  //密码长度n，子串长度m，
char s[N]; //子串，
int f[N][N], ne[N];     

int main() {
    //读入
    cin >> n >> s+1;
    m = strlen(s+1);

    //kmp的next数组处理,
    for (int i = 2, j = 0; i<=n; i++) {
        while (j != 0 && s[i] != s[j+1]) j = ne[j];
        if (s[i] == s[j+1]) j++;
        ne[i] = j;
    }

    f[0][0] = 1;  //初始化，对于第0位密码，且匹配字串的位置为0时的方案数为1,
    
    for (int i = 0; i<n; i++) {                 //枚举密码长度,
        for (int j = 0; j<m; j++) {             //枚举密码长度为i，且子串s匹配的位置为j时的状态,
            for (char k = 'a'; k<='z'; k++) {   //对于第i+1位字母，枚举26种可能的情况,
                //当前密码位数为i，且子串匹配的位置是j，尝试用i+1为k的情况进行跳转,
                //1.如果u能够跳转到子串s的最后一位(u==m)，说明密码第i+1位为k时，密码中会包含子串s,
                //2.反之则说明当密码的i+1位为k时不会包含子串s (u<m)，也就是合法方案，可以进行更新,
                int u = j;
                while (u != 0 && k != s[u+1]) u = ne[u];
                if (k == s[u+1]) u++;
                if (u < m) {
                    f[i+1][u] = (f[i+1][u] + f[i][j]) % mod;
                }
            }
        }          
    }

    //求方案总和，这里不取到m是因为当匹配到子串的第m位时，说明当前方案包含了整个子串，显然是不合法的方案,
    int res = 0;
    for (int i = 0; i<m; i++) res = (res + f[n][i]) % mod;  
    cout << res;
    return 0;
}