bzoj 1030 文本生成器【AC自动机+dp】

解题思路：

题目要求至少含有一个单词的方案数，可以转化成总方案数（$26^m$）减去不含有单词的方案数。
接下来把单词插到AC自动机上进行dp。
f[i][j]表示走到AC自动机的j号结点当前单词长度为i的方案数。
转移就是枚举下一位的26个字母，看是否可行（如果下一位在j的子节点中出现，若被标记为单词的结尾，就不能转移）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#define ll long long
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int L=105,N=60,p=10007;
struct node
{
    int son[26],fail;
    bool end;
}tr[N*L];
int n,m,tot=1,f[L][N*L];
char s[L];
queue<int>q;

void Insert()
{
    int len=strlen(s+1),po=1;
    for(int i=1;i<=len;i++)
    {
        if(!tr[po].son[s[i]-'A'])tr[po].son[s[i]-'A']=++tot;
        po=tr[po].son[s[i]-'A'];
    }
    tr[po].end=true;
}

void build_fail()
{
    q.push(1);
    while(!q.empty())
    {
        int u=q.front();q.pop();
        for(int i=0;i<26;i++)
        {
            int v=tr[u].fail,w=tr[u].son[i];
            while(!tr[v].son[i])v=tr[v].fail;
            v=tr[v].son[i];
            if(w)tr[w].fail=v,q.push(w),tr[w].end=tr[w].end||tr[v].end;
            else tr[u].son[i]=v;
        }
    }
}

void solve()
{
    int ans=1;
    for(int i=1;i<=m;i++)ans=ans*26%p;
    f[0][1]=1;
    for(int i=0;i<m;i++)
    {
        for(int j=1;j<=tot;j++)
            if(!tr[j].end&&f[i][j])
            {
                for(int k=0;k<26;k++)
                {
                    int v=j;
                    while(!tr[v].son[k])v=tr[v].fail;
                    v=tr[v].son[k];
                    f[i+1][v]=(f[i+1][v]+f[i][j])%p;
                }
            }
    }
    for(int i=1;i<=tot;i++)
        if(!tr[i].end)ans=((ans-f[m][i])%p+p)%p;
    cout<<ans;
}

int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),m=getint();
    for(int i=0;i<26;i++)tr[0].son[i]=1;
    while(n--)
    {
        scanf("%s",s+1);
        Insert();
    }
    build_fail();
    solve();
    return 0;
}