【BZOJ 1030】[JSOI2007]文本生成器 Ac自动机+fail树上dp

这道题要简单一些，其实我想提的是大白在介绍Ac自动机的时候引入的例题，刚刚看了一下，在P217 字串那道题，有相似之处，都是构造出fail树然后树上跑dp，唯一区别在于那道题是概率dp，很好的题目。

思想：首先，我们明确如果直接求有多上符合条件是非常麻烦的，所以我们采用逆向思维，求出所有不合法的方案然后用总的方案减去不合法的方案。至于怎么求不合法的方案，首先构造出fail树，然后当我们每随机生成一个字符的时候就类似与在树上向下走了一步（注意，当树上没有这个字符的时候就ch[u][i]==0就相当于回到根节点从新匹配，依旧合法），所以问题变成从一个节点出发走m步，其中不经过任何单词结尾节点的方案数（在构造树的时候处理）。定义f[i][j]表示现在在i节点已经走了j步的方案数，dp就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#define Mod 10007
#define maxn 6010
using namespace std;
int n,m,ch[maxn][26],f[maxn][maxn],fail[maxn],is[maxn],tot,q[maxn],ans=1;
char s[maxn];

void insert(){
	int len=strlen(s),rt=0;
	for(int x,i=0;i<len;i++){
		x=s[i]-'A';
		if(!ch[rt][x])ch[rt][x]=++tot;
		rt=ch[rt][x];
	}
	is[rt]=1;
}

void work(){
	int l=0,r=1;
	while(l<r){
		int u=q[l++];
		for(int v,i=0;i<26;i++){
			v=ch[u][i];
			if(v)q[r++]=v,fail[v]= u==0 ? 0 : ch[fail[u]][i];
			else ch[u][i]= u==0 ? 0 : ch[fail[u]][i];
			is[u]|=is[fail[u]];
		}
	}
}

void solve(){
	f[0][0]=1;
	for(int i=1;i<=m;i++){
		for(int j=0;j<=tot;j++){
			if(is[j])continue;
			for(int k=0;k<26;k++)
				f[ch[j][k]][i]=(f[ch[j][k]][i]+f[j][i-1])%Mod;
		}
	}
	for(int i=0;i<=tot;i++)if(is[i]==0)ans=(ans+Mod-f[i][m])%Mod;
	cout<<ans;
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)ans=ans*26%Mod;
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		insert();
	}
	work();
	solve();
	return 0;
}