AcWing1052提高课dp系列状态机模型

这篇博客探讨了一种利用状态机和KMP算法来设计满足特定条件的密码的方法。题目要求生成一个长度为N的密码S,其中S不包含特定的子串T。通过建立f[i][j]的状态转移矩阵,表示已生成i位密码且第i位在模式串T中匹配到的位置为j的方案数。利用KMP算法找到下一个字符在T中的匹配位置,并进行状态转移。博客详细解释了状态转移的过程,并提供了相应的C++代码实现。最后,通过累加f[n][0~len-1]得到满足条件的密码方案总数,并对结果取模10^9 + 7。该方法展示了字符串处理和算法在解决复杂问题中的应用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

状态机 + kmp

你现在需要设计一个密码 S,S 需要满足:

S 的长度是 N;
S 只包含小写英文字母;
S 不包含子串 T;
例如:abc 和 abcde 是 abcde 的子串,abd 不是 abcde 的子串。

请问共有多少种不同的密码满足要求?

由于答案会非常大,请输出答案模 109+7 的余数。

输入格式
第一行输入整数N,表示密码的长度。

第二行输入字符串T,T中只包含小写字母。

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

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

输入样例1:
2
a
输出样例1:
625
输入样例2:
4
cbc
输出样例2:
456924

设f[i][j]表示当前已经生成密码的第i位,且第i位在模式串t中能匹配到的位置为j的方案数
在这里插入图片描述
这表示从s[i - j + 1]到 s[i] 与t[1] 到 t[j] 的这一段两个字符串相等,假设t的长度为len
那么当j==len时,生成的密码不合法,所以在状态转移的过程中,j不能转移到len

那么如何描述状态的转移呢
假设第i + 1位能匹配到在t中的位置为u,通过kmp算法,我们可以找到这个位置u
那么f[i][j]能转移到f[i + 1][u],所以从f[i][j]这一条边到f[i + 1][u]这条路径存在,
那么f[i + 1][u] += f[i][j]
在这里插入图片描述

在图中假设的是s[i + 1] == ‘a’的情况,在实际运行中我们需要考虑s[i + 1]是所有小写字母的情况
所以需要枚举s[i + 1]从’a’到’z’
最后的答案从定义上可知是f[n][0~len - 1]之和

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

const int N = 55, mod = 1e9 + 7;
int f[N][N];
char t[N];
int ne[N];
int n;

int main(){
    cin >> n;
    cin >> t + 1;
    int len = strlen(t + 1);
    
    // kmp模板
    for(int i = 2, j = 0; i <= n; i ++){
        while(j && t[i] != t[j + 1]) j = ne[j];
        if(t[i] == t[j + 1]) j ++;
        ne[i] = j;
    }
    f[0][0] = 1;//当前密码生成0位,在t中匹配位置为0,方案为一个(初始化)
    for(int i = 0; i < n; i ++){
        for(int j = 0; j < len; j ++){
            for(int k = 'a'; k <= 'z'; k ++){//枚举第i + 1位为'a'到'z'的情况
                int u = j;
                while(u && t[u + 1] != k) u = ne[u];
                if(k == t[u + 1]) u ++;//找到第i + 1位在t中所能匹配到的位置
                
                if(u < len) f[i + 1][u] = (f[i + 1][u] + f[i][j]) % mod;
                //答案需要对1e9+7取模
            }
        }
    }
    int res = 0;
    for(int i = 0; i < len; i ++) res = (res + f[n][i]) % mod;
    cout << res << endl;
    return 0;
    
}

感谢指正

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值