BZOJ 1030 【JSOI2007】文本生成器

1030: [JSOI2007]文本生成器

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 3397   Solved: 1391
[ Submit][ Status][ Discuss]

Description

　　JSOI交给队员ZYX一个任务，编制一个称之为“文本生成器”的电脑软件：该软件的使用者是一些低幼人群，
他们现在使用的是GW文本生成器v6版。该软件可以随机生成一些文章―――总是生成一篇长度固定且完全随机的文
章—— 也就是说，生成的文章中每个字节都是完全随机的。如果一篇文章中至少包含使用者们了解的一个单词，
那么我们说这篇文章是可读的（我们称文章a包含单词b，当且仅当单词b是文章a的子串）。但是，即使按照这样的
标准，使用者现在使用的GW文本生成器v6版所生成的文章也是几乎完全不可读的?。ZYX需要指出GW文本生成器 v6
生成的所有文本中可读文本的数量，以便能够成功获得v7更新版。你能帮助他吗？

Input

　　输入文件的第一行包含两个正整数，分别是使用者了解的单词总数N (<= 60)，GW文本生成器 v6生成的文本固
定长度M；以下N行，每一行包含一个使用者了解的单词。这里所有单词及文本的长度不会超过100，并且只可能包
含英文大写字母A..Z

Output

　　一个整数，表示可能的文章总数。只需要知道结果模10007的值。

Sample Input

2 2
A
B

Sample Output

100

HINT

Source

【题目分析】

    AC自动机+动态规划

f[i][j][0/1]表示i步到j，未经过单词/已经经过单词，这样就很好转移了

【代码】

#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int md=10007;
int fail[6001],trie[6001][26],n,m,tot=1,v[6001];
inline void insert()
{
	char s[101];
	scanf("%s",s);
	int now=1,len=strlen(s)-1;
	for (int i=0;i<=len;++i){
		if (!trie[now][s[i]-'A']) trie[now][s[i]-'A']=++tot;
		now=trie[now][s[i]-'A'];
	}
	v[now]++;
}
inline void bfs()
{
	queue<int> q;
	q.push(1);
	while (!q.empty())
	{
		int x=q.front(); q.pop();v[x]|=v[fail[x]];
		for (int i=0;i<=25;++i){
			int j=fail[x];
			while (j&&!trie[j][i]) j=fail[j];
			if (trie[x][i]) {fail[trie[x][i]]=j?trie[j][i]:1; q.push(trie[x][i]);}
			else trie[x][i]=j?trie[j][i]:1;
		}
	}
}
int f[101][6001][2];
inline void dp()
{
	f[0][1][0]=1;
	for (int i=0;i<m;++i){
		for (int j=1;j<=tot;++j){
			for (int k=0;k<=25;++k){
				for (int l=0;l<=1;++l){
					if (v[trie[j][k]]) f[i+1][trie[j][k]][1]+=f[i][j][l],f[i+1][trie[j][k]][1]%=md;
					else f[i+1][trie[j][k]][l]+=f[i][j][l],f[i+1][trie[j][k]][l]%=md;
				}
			}
		}
	}
	int ans=0;
	for (int i=1;i<=tot;++i){
		ans+=f[m][i][1];
		ans%=md;
	}
	printf("%d\n",ans);
}
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;++i) insert();
	bfs();
	dp();
	return 0;
}