【AC自动机+DP】BZOJ1030(JSOI2007)[文本生成器]题解

题目概述

如果一篇文章中至少包含给出 $n$ 个单词中的一个单词，那么我们说这篇文章是可读的。求长度为 $m$ 字符集只有 $[A,Z]$ 的文本中可读文本的数量。

解题报告

AC自动机+DP套路题……由于直接求可读文本比较困难，正难则反，我们求不可读文本的数目 $ans$ ，那么 $26^m-ans$ 就是答案。

先将 $n$ 个模板串建成AC自动机，然后 $f[i][j]$ 表示文本串长度为 $i$ ，目前在AC自动机 $j$ 节点上的方案数，DP时避开匹配节点。

第一次做犯了经典错误……忘记了“如果 $fail(x)$ 是匹配节点，那么 x <script type="math/tex" id="MathJax-Element-11">x</script> 也是匹配节点”的传递关系。

示例程序

#include<cstdio>
using namespace std;
const int maxl=100,maxt=6000,MOD=10007;

int n,m,f[maxl+5][maxt+5],ans;
char s[maxl+5];bool vis[maxt+5];

int si,son[maxt+5][26];
inline void Insert(char *s)
{
    int p=0;
    for (int i=1;s[i];i++)
    {
        if (!son[p][s[i]-'A']) son[p][s[i]-'A']=++si;
        p=son[p][s[i]-'A'];
    }
    vis[p]=true;
}
int que[maxt+5],fa[maxt+5];
void ACAM()
{
    int Head=0,Tail=0;for (int i=0;i<26;i++) if (son[0][i]) que[++Tail]=son[0][i];
    while (Head!=Tail)
    {
        int x=que[++Head];vis[x]|=vis[fa[x]];
        for (int i=0,u;i<26;i++) if (u=son[x][i])
            fa[u]=son[fa[x]][i],que[++Tail]=u; else son[x][i]=son[fa[x]][i];
    }
}
inline void AMOD(int &x,int tem) {if ((x+=tem)>=MOD) x-=MOD;}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d%d",&n,&m);for (int i=1;i<=n;i++) scanf("%s",s+1),Insert(s);
    ACAM();f[0][0]=1;
    for (int i=0;i<m;i++)
    for (int j=0;j<=si;j++) if (!vis[j])
    for (int k=0;k<26;k++) if (!vis[son[j][k]])
        AMOD(f[i+1][son[j][k]],f[i][j]);
    ans=1;for (int i=1;i<=m;i++) ans=ans*26%MOD;
    for (int j=0;j<=si;j++) if (!vis[j]) AMOD(ans,MOD-f[m][j]);
    return printf("%d\n",ans),0;
}