【BZOJ】【P1030】【JSOI2007】【文本生成器】【题解】【AC自动机】

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=1030

f[i][j][k]表示当前文本第i个,fail树节点第j个,未/已 匹配

f[i][j->ch][k|j->ch->val]+=f[i-1][j][k]

ans=\sum f[m][i][1]

Code:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int n,p=10007;
int ch[6010][27];
int val[6010],fail[6010];
int m,cnt=1;
char s[101];
queue<int>q;
int f[110][6010][2];
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%s",s);int len=strlen(s);
		int u=1;
		for(int j=0;j<len;++j){
			int v=s[j]-'A'+1;
			if(!ch[u][v])ch[u][v]=++cnt;
			u=ch[u][v];
		}val[u]=1;
	}q.push(1);
	for(int u;!q.empty();){
		u=q.front();q.pop();val[u]|=val[fail[u]];
		for(int i=1,v;i<=26;++i)if(v=ch[u][i]){
			int j=fail[u];q.push(v);
			while(j&&!ch[j][i])j=fail[j];
			fail[v]=j&&ch[j][i]?ch[j][i]:1;
		}else{
			int j=fail[u];
			while(j&&!ch[j][i])j=fail[j];
			ch[u][i]=j&&ch[j][i]?ch[j][i]:1;
		}
	}int ans=0;
	f[0][1][0]=1;
	for(int i=1;i<=m;++i)
	for(int j=1;j<=cnt;j++)
	for(int k=1;k<=26;k++)
	for(int l=0;l<2;l++){
		int x=val[ch[j][k]]?1:l;
		f[i][ch[j][k]][x]=(f[i][ch[j][k]][x]+f[i-1][j][l])%p;
	}for(int i=1;i<=cnt;i++)ans=(ans+f[m][i][1])%p;
	cout<<ans<<endl;
	return 0;
}