BZOJ1030 ||洛谷P4052 [JSOI2007]文本生成器【AC自动机+DP】

Time Limit: 1 Sec
Memory Limit: 162 MB

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的值。

---

题目分析

此题直接统计可行的文本不太方便，所以转换一下思路
ans=所有文本数-不可读的文本数

AC自动机的DP一般都很套路
$dp[i][j]$就表示完成文本前$i$位（从根出发走了i步），当前走到AC自动机上的结点$j$，共有多少种方案不经过单词结点
从父结点向子节点转移即可
$dp[i][ch[j][k]]+=dp[i][j]$，初始时$dp[0][0]=1$

递推写法代码如下

const int mod=10007;
const int maxn=10010;
int n,m,cnt,ans;
int ch[maxn][30],rem[maxn];
int fail[maxn],dp[105][maxn];
queue<int> q;
char pt[maxn];
  
int qpow(int a,int k)
{
    int res=1;
    while(k){
        if(k&1) res=(res*a)%mod;
        a=(a*a)%mod; k>>=1;
    }
    return res;
}
  
void ins(char* ss,int len)
{
    int u=0;
    for(int i=0;i<len;++i)
    {
        int x=ss[i]-'A';
        if(!ch[u][x]) ch[u][x]=++cnt;
        u=ch[u][x];
    }
    rem[u]=1;
}
  
void ACM()
{
    for(int i=0;i<26;++i)
    if(ch[0][i]) fail[ch[0][i]]=0,q.push(ch[0][i]);
  
    while(!q.empty())
    {
        int u=q.front(); q.pop();
        for(int i=0;i<26;++i)
        {
            if(!ch[u][i]) ch[u][i]=ch[fail[u]][i];
            else{
                fail[ch[u][i]]=ch[fail[u]][i];
                rem[ch[u][i]]|=rem[fail[ch[u][i]]];
                q.push(ch[u][i]);
            }
        }
    }
}
  
int DP()
{
    dp[0][0]=1;
    for(int i=1;i<=m;++i)
    for(int j=0;j<=cnt;++j)
    for(int k=0;k<26;++k)
    {
        if(rem[ch[j][k]]) continue;
        dp[i][ch[j][k]]+=dp[i-1][j];
        dp[i][ch[j][k]]%=mod;
    }
      
    for(int i=0;i<=cnt;++i)
    ans=(ans+dp[m][i])%mod;
    return ans;
}
  
int main()
{
    n=read(); m=read();
    for(int i=1;i<=n;++i)
    {
        scanf("%s",&pt);
        ins(pt,strlen(pt));
    }
    ACM(); DP();
    printf("%d",((qpow(26,m)-ans)%mod+mod)%mod);
    return 0;
}


当然深搜记忆化也是可以的
dp[u][k]表示从u出发走k步，有几种方案满足路径上不经过单词结点
边界条件dp[u][m]=1，答案dp[0][m]

lt dp[maxn][210],vis[maxn][210];

lt DP(int u,int k)
{
	if(k==0) return dp[u][k]=1;
	if(vis[u][k]) return dp[u][k];
	
	lt res=0;
	for(int i=0;i<26;++i)
	{
		if(rem[ch[u][i]]) continue;
		res+=DP(ch[u][i],k-1);
		res%=mod;
	}
	
	vis[u][k]=1;
	return dp[u][k]=res;
}

lt ans=qpow(26,m)-DP(0,m);
printf("%lld",(ans%mod+mod)%mod);