AcWing 1052. 设计密码
阅读原文时间:2022年04月04日阅读:1

//f[i][j]表示前 i 个字符与字符串匹配长度为 j 时的方案数
#include
#include
#include
#include
using namespace std;
const int N=,mod=1e9+;;
int ne[N];//KMP中的next数组
char str[N];//子串
int f[N][N];//表示对于枚举到的那个点的某个状态的解
int main() {
int n,m;
cin >> n >> str+;
m=strlen(str+);
for(int i=,j=; i<=m; i++) {
while(j&&str[i]!=str[j+])j=ne[j];
if(str[i]==str[j+])j++;
ne[i]=j;
}
f[][]=;
for(int i=; i<n; i++) { //枚举当前密码的长度
for(int j=; j<m; j++) { //枚举子串中的位置,也就是当前密码,与子串匹配的长度
for(char k='a'; k<='z'; k++) { //枚举i处的字符
int u=j;//k要和 str[u+1]匹配,所以从0开始
while(u&&k!=str[u+]) u=ne[u];
if(str[u+]==k)u++;
if(u<m) { //说明没有出现过子串
f[i+][u]=(f[i+][u]+f[i][j])%mod;//这里最后的状态的f[i+1][u],所以更新的是这个值
}
}
}
}
int res=;
for(int i=; i<m; i++)res = (res + f[n][i]) % mod; //把所有的情况加起来
cout << res;
}