【bzoj 1030】文本生成器（AC自动机+DP）

传送门biu~
要计算可读文本数量，可以先计算所有不可读的文本数量，即不能匹配到任何模式串的方案数，再用$26^m$减去即可。
将AC自动机上所有危险点标记，$f_{i,j}$表示匹配$j$个字母后刚好在AC自动机的$i$号节点上（不匹配危险点）的方案数。转移时，假设节点$i$和它的儿子$x$都不是危险的，那么$f_{x,j+1}+=f_{i,j}$（可以采用刷表法）。最后用$26^m$减去$\Sigma f_{i,m}$即可。

#include<bits/stdc++.h>
using namespace std;
const int mod=10007;char s[1005];
int n,m,cnt,ans,f[100005][150];
struct Node{
    Node *ch[26],*nex;
    bool b;int num;
    Node();
}*root=new Node,*pre[100005];
Node :: Node(){
    nex=NULL;b=false;
    for(int i=0;i<26;++i)   ch[i]=NULL;
    num=++cnt;pre[cnt]=this;
}
inline void Insert(char s[]){
    Node *x=root;
    for(int i=1;s[i];++i){
        int to=s[i]-'A';
        if(!x->ch[to])  x->ch[to]=new Node;
        x=x->ch[to];
    }
    x->b=true;
}
queue<Node*>q;
inline void GetFail(){
    root->nex=root;
    for(int i=0;i<26;++i){
        if(root->ch[i]) q.push(root->ch[i]),root->ch[i]->nex=root;
        else    root->ch[i]=root;
    }
    while(!q.empty()){
        Node *x=q.front();q.pop();
        for(int i=0;i<26;++i){
            if(x->ch[i])    q.push(x->ch[i]),x->ch[i]->nex=x->nex->ch[i];
            else    x->ch[i]=x->nex->ch[i];
        }
        Node *tmp=x;
        while(tmp!=root && !tmp->b) tmp=tmp->nex;
        if(tmp->b)  x->b=true;
    }
}
inline void dp(){
    f[root->num][0]=1;
    for(int i=0;i<m;++i){
        for(int j=1;j<=cnt;++j){
            if(pre[j]->b || !f[j][i])   continue;
            for(int k=0;k<26;++k){
                if(pre[j]->ch[k]->b)    continue;
                (f[pre[j]->ch[k]->num][i+1]+=f[j][i])%=mod;
            }
        }
    }
    for(int i=1;i<=cnt;++i) ans=(ans+f[i][m])%mod;
}
inline int ksm(int p,int n){
    int re=1;
    while(n){
        if(n&1) re=re*p%mod;
        p=p*p%mod;
        n>>=1;
    }
    return re;
}
int main(){
    scanf("%d%d",&n,&m);
    while(n--)  scanf("%s",s+1),Insert(s);
    GetFail();dp();
    ans=(ksm(26,m)-ans+mod)%mod;
    printf("%d\n",ans);
    return 0;
}