[Acwing] 1052. 设计密码 状态机模型

前言

传送门 :

感觉这个题不应该分类到状态机里面啊

思路

状态表示 :
$f[i][j]$到第$i$位,且匹配到$j$的方案数

状态转移 :
每次通过$KMP$的$ne$数组进行转移

Mycode

const int N  = 55 , mod = 1e9+7;
int f[N][N] , ne[N];
//已经生成了i位,并且匹配到了j位
char str[N];

void solve(){
	int n,m;
	cin>>n>>str+1;
	m = strlen(str+1);
	
	
	//用来求ne数组
	for(int i=2,j=0;i<=m;i++){
		while(j&&str[j+1]!=str[i]) j = ne[i];
		if(str[j+1] == str[i]) j++;
		ne[i] = j ;
	}
	f[0][0] = 1;
	//已经匹配到 
	for(int i=1;i<=n;i++) //枚举密码位
		for(int j = 0;j<m;j++) //把第i位密码匹配到 子串位置都枚举
			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++;
				if(u<m)
				f[i][u] = (f[i][u] + f[i-1][j])%mod;
			}
			
	int res = 0;
	for(int i=0;i<m;i++){
		res = (res + f[n][i])%mod;
	}
	cout<<res<<endl;
	
}