【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;
}