bzoj 1030: [JSOI2007]文本生成器 (ac自动机上的dp)

1030: [JSOI2007]文本生成器

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

ac自动机上的dp
dp[0][i][j]表示到第i位，以第j个状态结束不符合条件的串的个数
dp[1][i][j]表示到第i位，以第j个状态结束符合条件的串的个数

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 6005;
const int MOD = 10007;

int que[N], fr, ta;

struct ACM {
	int cnt;
	int nxt[N][26], sum[N], fail[N];
	void init() {
		for(int i = 1; i <= cnt; ++i) {
			sum[i] = fail[i] = 0;
			for(int j = 0; j < 26; ++j)
				nxt[i][j] = 0;
		}
		cnt = 1;
		for(int i = 0; i < 26; ++i) nxt[0][i] = 1;
	}
	void insert(string str) {
		int now = 1;
		int len = str.length();
		for(int i = 0; i < len; ++i) {
			if(nxt[now][str[i] - 'A'] == 0) nxt[now][str[i] - 'A'] = ++cnt;
			now = nxt[now][str[i] - 'A'];
		}
		sum[now] = 1;
	}
	void build_fail() {
		fr = ta = 0;
		que[ta++] = 1;
		fail[1] = 0;
		while(fr != ta) {
			int now = que[fr++];
			for(int i = 0; i < 26; ++i) {
				int x = nxt[now][i];
				if(x == 0) continue;
				int tmp = fail[now];
				while(nxt[tmp][i] == 0) tmp = fail[tmp];
				fail[x] = nxt[tmp][i];
				que[ta++] = x;
			}
		}
	}
	void debug() {
		for(int i = 1; i <= cnt; ++i) {
			cout<<i<<": "<<"fail = "<<fail[i]<<" [ ";
			for(int j = 0; j < 26; ++j) {
				if(nxt[i][j]) cout<<char(j + 'A')<<','<<nxt[i][j]<<' ';
			}
			puts("]");
		}
	}
	void gao(int, int);
} acm;

int dp[2][105][N];

void ACM::gao(int n, int m) {
	build_fail();
	//debug();
	memset(dp, 0, sizeof(dp));
	dp[0][0][1] = 1;
	for(int i = 1; i <= m; ++i) {
		for(int j = 1; j <= cnt; ++j) {
			int u = j;
			for(int k = 0; k < 26; ++k) {
				int v = u;
				while(nxt[v][k] == 0) v = fail[v];
				v = nxt[v][k];
				if(sum[v]) {
					dp[1][i][v] = (dp[1][i][v] + dp[1][i - 1][u] + dp[0][i - 1][u]) % MOD;
					dp[0][i][v] += 0;
				} else {
					dp[1][i][v] = (dp[1][i][v] + dp[1][i - 1][u]) % MOD;
					dp[0][i][v] = (dp[0][i][v] + dp[0][i - 1][u]) % MOD;
				}
			}
		}
	}
	int ans = 0;
	for(int i = 1; i <= cnt; ++i)
		ans = (ans + dp[1][m][i]) % MOD;
	cout<<ans<<endl;
}

string str[65];

bool ok(int x, int n) {
	int len = str[x].length();
	for(int i = 0; i < n; ++i) {
		if(i == x) continue;
		int pos = str[x].find(str[i]);
		if(pos >= 0 && pos < len) return false;
	}
	return true;
}

int main() {
	int n, m;
	cin>>n>>m;
	acm.init();
	for(int i = 0; i < n; ++i) {
		cin>>str[i];
	}
	sort(str, str + n);
	n = unique(str, str + n) - str;
	for(int i = 0; i < n; ++i) {
		if(ok(i, n)) acm.insert(str[i]);
	}
	acm.gao(n, m);
	return 0;
}