【bzoj1030】文本生成器 AC自动机+树型dp

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=1030 

【题解】

首先求出不可能的文章总数，然后减一下就行了。

建一个AC自动机，然后在AC自动机上做dp，f[i][j]表示当前长度为i，匹配到AC自动机上第j个结点时的方案数。

则f[i][temp]=f[i][temp]+f[i-1][j]  (temp是j结点往后匹配到的下一个结点)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAXN 110
#define mod 10007
struct node{int v,fail,Link[26];}tr[MAXN*60];
int n,m,len,ans,mul(1),q[MAXN*60],f[MAXN][MAXN*60];
char ch[MAXN];
inline int read()
{
    int x=0,f=1;  char ch=getchar();
    while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getchar();}
    while(isdigit(ch))  {x=x*10+ch-'0';  ch=getchar();}
    return x*f;
}
void insert(int k,int x)
{
    int dex=ch[k]-'A';
    if(!tr[x].Link[dex]) tr[x].Link[dex]=++len;
    int y=tr[x].Link[dex];
    if(k==strlen(ch+1))  {tr[y].v++;  return;}
    insert(k+1,y);
}
void build()
{
    int head=0,tail=1;
    while(++head<=tail)
    {
        int x=q[head];
        for(int i=0;i<26;i++)
            if(tr[x].Link[i])
            {
                int y=tr[x].Link[i];
                q[++tail]=y;
                if(!x) continue;
                int temp=tr[x].fail;
                while(temp&&!tr[temp].Link[i])  temp=tr[temp].fail;
                tr[y].fail=tr[temp].Link[i];
                if(tr[tr[temp].Link[i]].v) tr[y].v=1;
            }
    }
}
void dp()
{
    f[0][0]=1;
    for(int i=1;i<=m;i++)
        for(int j=0;j<=len;j++)
            if(f[i-1][j]&&(!tr[j].v))
                for(int k=0;k<26;k++)
                {
                    int temp=j;
                    while(temp&&!tr[temp].Link[k]) temp=tr[temp].fail;
                    temp=tr[temp].Link[k];
                    if(!tr[temp].v)  f[i][temp]=(f[i][temp]+f[i-1][j])%mod;
                }
}
int main()
{
    //freopen("cin.in","r",stdin);
    //freopen("cout.out","w",stdout);
    n=read();  m=read();
    for(int i=1;i<=n;i++){scanf("%s",ch+1);insert(1,0);}
    build();
    dp();
    for(int i=0;i<=len;i++)  ans=(ans+f[m][i])%mod;
    for(int i=1;i<=m;i++)  mul=(mul*26)%mod;
    printf("%d\n",(mul-ans+mod)%mod);
    return 0;
}