【bzoj1030】: [JSOI2007]文本生成器 字符串-AC自动机-DP

【bzoj1030】: [JSOI2007]文本生成器

首先把匹配任意一个的个数的问题转化为总个数-没有一个匹配的个数

先构造AC自动机，然后枚举每一位的字母以及在自动机上的位置

f[i][j]为第i位在j的位置且没有匹配过任何一个串的个数

然后26^m-sum(f[m][j])就是答案

还有就是当p->fail一直到root的路径上只要有一个点是一个串的终点那么点f[i][p]就要ban掉 因为这个WA了好多次

/* http://www.cnblogs.com/karl07/ */
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

#define P 10007
struct trie{
    trie *next[26],*fail;
    int th,w;
}t[6005],*root=t,*NEW=t,*q[6005];

int n,m,l,r,ans;
int f[105][6005];
char s[105];

trie *new1(){NEW++; NEW->th=NEW->w=0; return NEW;}

#define pnf p->next[i]->fail
#define pn p->next[i]
trie *insert(trie *p,int i,int w){
    if (!pn) pn=new1();
    pn->w|=w;
    return pn;
}

void build_fail(){
    trie *p=q[0]=root;
    for (int i=0;i<26;i++) if (pn) pnf=p,q[++r]=pn,pn->th=r;
    while (l<r){
        p=q[++l];
        for (int i=0;i<26;i++){
            if (pn){
                q[++r]=pn,pn->th=r;
                for (pnf=p->fail ; pnf!=root && !pnf->next[i] ; pnf=pnf->fail);
                if (pnf->next[i]) pnf=pnf->next[i];
                pn->w|=pnf->w;
            }
        }
    }
}

int Q_pow(int x,int y){
    int ans=1;
    for (;y; x=x*x%P , y=y>>1 ) if (y&1) ans=ans*x%P;
    return ans;
}
void dp(){
    f[0][0]=1;
    for (int k=0;k<m;k++){
        for (int j=0;j<=r;j++) if (f[k][j] && !q[j]->w){
            for (int i=0;i<26;i++){
                trie *p=q[j];
                while (!pn && p!=root) p=p->fail;
                if (pn) p=pn;
                f[k+1][p->th]=(f[k+1][p->th]+f[k][j])%P;
            }
        }
    }
    for (int i=0;i<=r;i++) if (!q[i]->w) ans=(ans+f[m][i])%P;
    printf("%d\n",(Q_pow(26,m)-ans+P)%P);
}
#undef pn
#undef pnf

int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++){
        scanf("%s",s);
        int j=0,l=strlen(s);
        for (trie *p=root;j<l;j++) p=insert(p,s[j]-'A',j==l-1);
    }
    build_fail();
    dp();
    return 0;
}