修复密码(kmp+状态机dp)

构造的密码位置i和j+1进行匹配,如果i确定,j+1在kmp中的转移是确定的。f[i][j]表示密码构造到i位置,状态是j的方案,枚举i+1位置处放哪个字母,固定后,就可以知道当前状态可以连向哪些状态。dp的复杂度NM26,因为里面现找它的转移到的状态,所以应该再乘M,复杂度为26*N^3

#include<bits/stdc++.h>

using namespace std;

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

int n,m;
char str[N];
int f[N][N];

int main()
{
	cin>>n>>str+1;
	m=strlen(str+1);
	
	int ne[N]={0};
	for(int i=2,j=0;i<=m;i++)
	{
		while(j&&str[i]!=str[j+1]) j=ne[j];
		if(str[i]==str[j+1]) j++;
		ne[i]=j;
	}
	
	f[0][0]=1;
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			for(char k='a';k<='z';k++)
			{
				int u=j;
				while(u&&str[u+1]!=k) u=ne[u];
				if(str[u+1]==k) u++;
				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;
}
全部评论

相关推荐

10-25 12:05
已编辑
湖南科技大学 Java
若梦难了:我有你这简历,已经大厂乱杀了
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务