bzoj 1030 文本生成器（AC自动机+dp）

版权声明：版权声明：本文为博主原创文章，未经博主允许不得转载，欢迎添加友链。 https://blog.csdn.net/zzk_233/article/details/82854419

前置知识ac自动机或者trie图。

根据题意需要求出文章中包含1个或更多的字符串的数量。但是这明显很难求，所以求出所有方案减去不包含的数量。

考虑dp，设f[i][j]表示第i位置时为AC自动机（trie图）上第j个字符没有匹配的方案数，如果这个字符的下一个字符没有匹配就可以

转移，预处理那些字符可以匹配只需在求fail指针的时候继承一下fail指针的状态即可。

我写的trie图

代码

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<queue>
#define mode 10007
using namespace std;
char ss[65][150];
int t[60105][27];
int fail[60105];
int f[105][60105];
int fin[60105];
int l,sz;
void insert(char *s)
{
	l=strlen(s);
	int root=0,k;
	for(int i=0;i<l;i++)
	{
		k=s[i]-'A';
		if(!t[root][k])t[root][k]=++sz;
		root=t[root][k];
	}
	fin[root]++;
}
queue<int>M;
void get_fail()
{
	int root=0;
	for(int i=0;i<26;i++)
	{
		if(t[root][i])M.push(t[root][i]);
	}
	while(!M.empty())
	{
		root=M.front();M.pop();
		for(int i=0;i<26;i++)
		{
			if(t[root][i])
			{
				fail[t[root][i]]=t[fail[root]][i];
				fin[t[root][i]]+=fin[t[fail[root]][i]];
				M.push(t[root][i]);
			}else
			{
				t[root][i]=t[fail[root]][i];
			}
		}
	}
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%s",ss[i]);
		insert(ss[i]);
	}
	get_fail();
	f[0][0]=1;
	for(int i=1;i<=m;i++)
	{
		for(int j=0;j<=sz;j++)
		{
			for(int k=0;k<26;k++)
			{
				if(!fin[t[j][k]])
				{
					f[i][t[j][k]]+=f[i-1][j];
					f[i][t[j][k]]%=mode;
				}
			}
		}
	}
	int ans=0;
	for(int i=0;i<=sz;i++)
	{
		ans+=f[m][i];
		ans%=mode;
	}
	int sum=1;
	for(int i=1;i<=m;i++)
	{
		sum=sum*26%mode;
	}
	printf("%d",((sum-ans)%mode+mode)%mode);
	return 0;
}