【BZOJ】1030 [JSOI2007]文本生成器 AC自动机+DP

题目传送门

经过这道题的练习，我对AC自动机有了更加深刻的理解：怎么计算AC自动机需要的数组大小。

（话说为什么我以前不会？大概是以前太瓜皮了吧……）

这题的题目大意就是让我们求构成长度为m的文本串，且文本串内至少包含一个模式串的方案总数。

我们可以用集合的思想，把题目转化成构成长度为m的文本串的方案数减去构成长度为m的文本串中不包含任意一个模式串的方案数。

然后考虑AC自动机，我们把每一个模式串插入Trie树中，标记每一个模式串结尾出的叶节点，建立next树。

接着我们考虑DP，定义f[i][j]表示当前文本串长度为i，所在Trie树上的节点为j的构成方案数，于是题目就转化成了求Σf[m][i]且节点i没有被标记过。

对于所有非叶子节点j，我们可以直接把f[i][j]转移到f[i+1][j的儿子]，注意：这里的j的儿子是经过指针优化的。

（“指针”这一说法只是我的个人喜好，具体的请看我的博客：【洛谷】3808 【模板】AC自动机（简单版））

至于时间复杂度，是O(size*m*26)的（size表示AC自动机的大小），反正能跑过这一题就行了。

附上AC代码：

#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;

const int p=10007;
struct note{
	int lk[26],ed,nt;
}AC[6000];
int n,m,size,f[105][6000],ans;
char s[105];

inline void insert(char *s){
	int len=strlen(s+1),now=0;
	for (int i=1; i<=len; ++i){
		if (!AC[now].lk[s[i]-'A']) AC[now].lk[s[i]-'A']=++size;
		now=AC[now].lk[s[i]-'A'];
	}
	AC[now].ed=1;
}

inline void build(){
	queue <int> que;
	for (int i=0; i<26; ++i)
		if (AC[0].lk[i]) AC[AC[0].lk[i]].nt=0,que.push(AC[0].lk[i]);
	while (!que.empty()){
		int d=que.front();que.pop();
		for (int i=0; i<26; ++i)
			if (AC[d].lk[i]) AC[AC[d].lk[i]].nt=AC[AC[d].nt].lk[i],que.push(AC[d].lk[i]);
				else AC[d].lk[i]=AC[AC[d].nt].lk[i];
		AC[d].ed|=AC[AC[d].nt].ed;
	}
	return;
}

int main(void){
	scanf("%d%d",&n,&m);
	for (int i=1; i<=n; ++i) scanf("%s",s+1),insert(s);
	AC[0].nt=0,build();
	f[0][0]=1;
	for (int i=1; i<=m; ++i)
		for (int j=0; j<=size; ++j)
			if (!AC[j].ed)
				for (int k=0; k<26; ++k)
					f[i][AC[j].lk[k]]=(f[i][AC[j].lk[k]]+f[i-1][j])%p;
	ans=1;
	for (int i=1; i<=m; ++i) ans=(ans*26)%p;
	for (int i=0; i<=size; ++i) if (!AC[i].ed) ans=(ans-f[m][i]+p)%p;
	printf("%d",ans);
	return 0;
}