BZOJ 1030: [JSOI2007]文本生成器 AC自动机

给出N个由大写字母组成的字符串，再给出一个长度M
求有多少个由大写字母组成的长度为M的字符串满足“至少包含这N个字符串中的一个”
A包含B指的是，B是A的子串
答案对10007取模

N<=60,每个字符串的长度<=100,M<=100

解析：这道题正着想显然是有点难，那我们就倒过来，用所有可能的文章总数减去不包含给定字符串的数量。
设F[i][j]表示匹配前i个字母之后走到了AC自动机的j号节点的方案数
则有对于任意的j的儿子节点x
若x不是危险节点，则F[i+1][x]+=F[i][j]
最后把所有不是危险节点的F[M][x]求和即是不包含这N个字符串的串的数量
用26^M减去即可

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<iostream>
#include<iomanip>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
const int mod=10007;
struct Trie
{
    Trie *son[26],*fail;
    bool match;
    int num;
    Trie()
    {
        memset(son,0,sizeof(son));
        fail=NULL;
        match=false;
    }
}mempool[50000],*root=&mempool[0];
int top=0;
void my_insert(char *s)
{
    Trie *o=root;
    while(*s)
    {
        if(!o->son[*s-'A']) o->son[*s-'A']=&mempool[++top];
        o=o->son[*s-'A'];
        s++;
    }
    o->match=true;
}
void bfs()
{
    static Trie* dui[50000];
    int top=1,my_final=1;
    for(int i=0;i<26;i++)
    {
        if(root->son[i])
        {
            root->son[i]->fail=root;
            dui[my_final++]=root->son[i];
        }
        else root->son[i]=root;
    }
    while(top<my_final)
    {
        Trie *o=dui[top];
        for(int i=0;i<26;i++)
        {
           if(o->son[i])
           {
                o->son[i]->fail=o->fail->son[i];
                dui[my_final++]=o->son[i];
           }
           else o->son[i]=o->fail->son[i];
        }
        top++;
        Trie *t=o->fail;
        while(t!=root && !t->match) t=t->fail;
        o->fail=t;
        if(t->match) o->match=true;
    }
}
int f[50000][101];
char s[10000];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        my_insert(s);
    }
    bfs();
    for(int i=0;i<=top;i++) mempool[i].num=i;
    f[0][0]=1;
    int ans=0;
    for(int k=0;k<m;k++)
        for(int i=0;i<=top;i++)
        {
            if(mempool[i].match) continue;
            for(int j=0;j<26;j++)
            {
                if(!mempool[i].son[j]->match)
                    f[mempool[i].son[j]->num][k+1]=(f[mempool[i].son[j]->num][k+1]+f[i][k])%mod;
            }
        }
    for(int i=0;i<=top;i++) ans=(ans+f[i][m])%mod;
    int t=m;
    int mid=26;
    int trueans=1;
    while(t)
    {
        if(t&1) trueans=(trueans*mid)%mod;
        mid=(mid*mid)%mod;
        t>>=1;
    }
    cout<<(trueans-ans+mod)%mod;
    return 0;
}