JSOI 2007 文本生成器 【AC自动机，树形DP】

JSOI 2007 文本生成器

题目见链接。

题解：

建立在 AC自动机 上的 树形DP。

先把单词建立 trie 树，并标记是否为一个单词的结尾(bool : ed)， 预处理 fail 指针， ed |= fail.ed

然后求 可读，可以用 总的 减去 不可读的

设 f[i][j] 表示 到了第 i 位，第 i 位上的节点编号为 j 的方案数，则

f[i][son[j]] = (f[i-1][j] + f[i][son[j]] ) % mo 。

最后用 ans - f[m][i] ( i 节点不为单词末尾)

 

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=6005,M=105,mo=10007;
char s[N];
int rt,tot,q[N],head,tail,ans1=1,ans2,f[M][N],n,m;
struct node{
    int fail,nxt[26];
    bool ed;
}e[N];
inline void insert(char *s)
{
    rt=0;
    for (int i=0; s[i]; ++i)
    {
        if (!e[rt].nxt[s[i]-'A'])
          e[rt].nxt[s[i]-'A']=++tot;
        rt=e[rt].nxt[s[i]-'A'];
    }
    e[rt].ed=1;
}
inline void getfail()
{
    rt=0; head=tail=0;
    for (int i=0; i<26; ++i)
      if (e[rt].nxt[i]) q[++tail]=e[rt].nxt[i];
    while (head<tail)
    {
        int x=q[++head];
        for (int i=0; i<26; ++i)
          if (e[x].nxt[i])
          {
                e[e[x].nxt[i]].fail=e[e[x].fail].nxt[i];
                q[++tail]=e[x].nxt[i];
          }
          else e[x].nxt[i]=e[e[x].fail].nxt[i];
        e[x].ed|=e[e[x].fail].ed;
    }
}
inline void doing()
{
    f[0][0]=1;
    for (int i=1; i<=m; ++i)
      for (int j=0; j<=tot; ++j)
      {
          if (e[j].ed) continue;
          for (int k=0; k<26; ++k)
            f[i][e[j].nxt[k]]=(f[i][e[j].nxt[k]]+f[i-1][j])%mo;
      }
    for (int i=0; i<=tot; ++i)
      if (!e[i].ed) ans2+=f[m][i];
    for (int i=1; i<=m; ++i)
      ans1=(ans1*26)%mo;
    printf("%d
",((ans1-ans2)%mo+mo)%mo);
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; ++i)
      scanf("%s",s),insert(s);
    getfail();
    doing();
    return 0;
}

fighting fighting fighting!!!