[AC自动机+DP] 限定长度不包含给定子串的可能字符串数量 BZOJ1030

1030: [JSOI2007]文本生成器

Time Limit: 1 Sec  Memory Limit: 162 MB
Submit: 6242  Solved: 2656
[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

 

 

#include <bits/stdc++.h>
using namespace std;

const int mn = 60 * 100 + 10;
const int mod = 10007;

int dp[105][mn];

int num;
int nx[mn][26], fail[mn];
bool ed[mn];
void add(char a[])
{
	int r = 0;
	int len = strlen(a);
	for (int i = 0; i < len; i++)
	{
		if (nx[r][a[i] - 'A'] == -1)
			nx[r][a[i] - 'A'] = ++num;
		r = nx[r][a[i] - 'A'];
	}
	ed[r] = 1;
}
void build_Fail()
{
	queue<int>que;
	int root = 0;
	fail[root] = root;
	for (int i = 0; i < 26; i++)
	{
		if (nx[root][i] == -1)
			nx[root][i] = root;
		else
		{
			fail[nx[root][i]] = root;
			que.push(nx[root][i]);
		}
	}
	while (!que.empty())
	{
		int now = que.front();
		que.pop();
		
		if (ed[fail[now]])  /// *****
			ed[now] = 1;
			
		for (int i = 0; i < 26; i++)
		{
			if (nx[now][i] == -1)
				nx[now][i] = nx[fail[now]][i];
			else
			{
				fail[nx[now][i]] = nx[fail[now]][i];
				que.push(nx[now][i]);
			}
		}
	}
}

int DP(int m) /// 第i步跳到j节点的可能情况数量
{
	dp[0][0] = 1;
	for (int i = 0; i < m; i++)
	{
		for (int j = 0; j <= num; j++)
		{
			for (int k = 0; k < 26; k++)
			{
				if (ed[nx[j][k]])
					continue;
				(dp[i + 1][nx[j][k]] += dp[i][j]) %= mod;
			}
		}
	}

	int ans = 0;
	for (int i = 0; i <= num; i++)
		(ans += dp[m][i]) %= mod;
	return ans;
}
int main()
{
	memset(nx, -1, sizeof nx);

	int n, m; 
	char ch[110];
	scanf("%d %d", &n, &m);
	while (n--)
	{
		scanf("%s", ch);
		add(ch);
	}
	build_Fail();

	int temp = DP(m);
	int res = 1;
	for (int i = 1; i <= m; i++)
		(res *= 26) %= mod;
	printf ("%d\n", (res - temp + mod) % mod);
	/// 总情况 - 不包含
	return 0;
}