bzoj 1030: [JSOI2007]文本生成器

当时道AC自动机的模板题吧。好久没打忘得七七八八。

AC自动机其实跟KMP没什么巨大区别。

这题直接做有些困难，考虑补集转化思想，即求不可阅读的方案数。

题意可以转化为在自动机上走m步，不经过单词结尾点（fail指针指向的点也算）

f[i][j]表示前i个字符走到j号点的方案数。

这样就很好转移了。

code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int mod=10007;
struct trnode{
	int a[27],fail;
}tr[6010];int root=1,tot=1;
int ans2=1,ans1=0;
int f[110][6010];
bool check[6010];
int n,m;
char s[110];
void build()
{
	int len=strlen(s+1);
	int x=root;
	for(int i=1;i<=len;i++)
	{
		int c=s[i]-'A';
		if(tr[x].a[c]) x=tr[x].a[c];
		else x=tr[x].a[c]=++tot;
	}
	check[x]=true;
}
queue<int> q;
void make_fail()
{
	q.push(root);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(int i=0;i<26;i++)
		{
			if(!tr[x].a[i]) continue;
			if(x==root) tr[tr[x].a[i]].fail=root;
			else
			{
				
				int j=tr[x].fail;
				while(j!=root&&!tr[j].a[i]) j=tr[j].fail;
				tr[tr[x].a[i]].fail=tr[j].a[i];
				if(check[tr[j].a[i]]) check[tr[x].a[i]]=true;
			}
			q.push(tr[x].a[i]);
		}
	}
}
void dp(int x)
{
	for(int i=1;i<=tot;i++)
	{
		if(check[i]||!f[x-1][i]) continue;
		for(int j=0;j<26;j++)
		{
			int k=i;
			while(!tr[k].a[j]) k=tr[k].fail;
			f[x][tr[k].a[j]]=(f[x][tr[k].a[j]]+f[x-1][i])%mod;
		}
	}
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=0;i<26;i++) tr[root].a[i]=++tot;
	for(int i=1;i<=n;i++)
	{
		scanf("%s",s+1);
		build();
	}
	make_fail();
	f[0][1]=1;
	for(int i=1;i<=m;i++) dp(i);
	for(int i=1;i<=m;i++)
	   ans2=(ans2*26)%mod;
    for(int i=1;i<=tot;i++)
       if(!check[i]) ans1=(ans1+f[m][i])%mod;
    printf("%d",((ans2-ans1)%mod+mod)%mod);
}