BZOJ.1030.[JSOI2007]文本生成器(AC自动机 DP)

题目链接

直接在AC自动机上DP。
\(f[i][j]\)表示 当前是文本第 \(i\)位，匹配到了AC自动机上的 \(j\)节点 此时的文章数。
转移时就枚举26个字母，转移到AC自动机上的下一个节点。
任意匹配一个单词就可以，所以当前节点 \(j\) 如果是模式串的结束节点，不再跳到下一个节点，而是直接*26给 \(f[i+1][j]\)。
不是结束节点的话 跳到下一个节点直接加上。

注意一个节点如果能通过fail指针跳到模式串的结束节点，那么它也是一个结束节点。。已经忘了。
容斥去算不满足条件的文本数量也可以。不过我这么做好像常数更优233.

//3900kb    68ms
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define mod (10007)
#define gc() getchar()
const int N=6e3+7;

int n,num,tot,f[103][N];
char s[105];
struct AC_Automaton
{
    int tot,son[N][26],fail[N],q[N];
    bool val[N];

    void Insert(char *s)
    {
        int l=strlen(s),x=0;
        for(int id,i=0; i<l; ++i)
        {
            if(!son[x][id=s[i]-'A']) son[x][id]=++tot;;
            x=son[x][id];
        }
        val[x]=1;
    }
    void Build()
    {
        int h=0,t=0;
        for(int i=0; i<26; ++i)
            if(son[0][i]) fail[son[0][i]]=0, q[t++]=son[0][i];
        while(h<t)
        {
            int x=q[h++];
            val[x]|=val[fail[x]];//!
            for(int v,i=0; i<26; ++i)
                if(son[x][i]) fail[son[x][i]]=son[fail[x]][i], q[t++]=son[x][i];
                else son[x][i]=son[fail[x]][i];
        }
    }
}ac;

int main()
{
    scanf("%d%d",&num,&n);
    while(num--) scanf("%s",s), ac.Insert(s);
    ac.Build();

    f[0][0]=1, tot=ac.tot;
    for(int i=0; i<n; ++i)
        for(int j=0; j<=tot; ++j)
            if(f[i][j]){
                if(ac.val[j]) f[i+1][j]+=f[i][j]*26%mod;//%mod
                else{
                    for(int k=0; k<26; ++k)
                        (f[i+1][ac.son[j][k]]+=f[i][j])%=mod;
                }
            }
    int Ans=0;
    for(int i=0; i<=tot; ++i)
        if(ac.val[i]) Ans+=f[n][i];
    printf("%d",Ans%mod);

    return 0;
}