bzoj1030:[JSOI2007]文本生成器

传送门

AC自动机上跑dp,首先想到正难则反
所以我们考虑不可读的字符串
\(f[i][j]\)为长度为\(i\),当前处理到第\(j\)个节点时的不可读字符串数
转移就很显然了
提示:如果一个节点的fail指针指向的节点中有一个是可读字符串的结尾节点,那么这个节点也是不可用的
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=1010,mod=10007;
queue<int>q;bool ed[6010];
int n,m,rt=1,id=1,ch[6010][26],ans=1,fail[6010],f[110][6010];char s[110];
void insert(char *s)
{
	int len=strlen(s+1);rt=1;
	for(rg int i=1;i<=len;i++)
	{
		int now=s[i]-'A';
		if(!ch[rt][now])ch[rt][now]=++id;
		rt=ch[rt][now];
	}
	ed[rt]=1;
}
void bfs()
{
	q.push(1);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(rg int i=0;i<26;i++)
		{
			if(!ch[x][i]){ch[x][i]=fail[x]?ch[fail[x]][i]:1;continue;}
			int j=fail[x],z=ch[x][i];q.push(z);
			while(j&&!ch[j][i])j=fail[j];
			if(j)fail[z]=ch[j][i];
			else fail[z]=1;
			ed[z]|=ed[fail[z]];
		}
	}
}
int main()
{
	read(n),read(m);
	for(rg int i=1;i<=n;i++)scanf("%s",s+1),insert(s);
	bfs();for(rg int i=1;i<=m;i++)ans=1ll*ans*26%mod;
	f[0][1]=1;
	for(rg int i=1;i<=m;i++)
		for(rg int j=1;j<=id;j++)
			for(rg int k=0;k<26;k++)
				if(!ed[ch[j][k]])
					f[i][ch[j][k]]=(f[i][ch[j][k]]+f[i-1][j])%mod;
	for(rg int i=1;i<=id;i++)ans=(ans-f[m][i]+mod)%mod;
	printf("%d\n",ans);
}
posted @ 2019-02-26 21:57  蒟蒻--lichenxi  阅读(71)  评论(0编辑  收藏