luogu P4052 [JSOI2007]文本生成器

背景：

暑假集训七月份最后一天，写写原来没有写过的$\text{blog}$吧。

题目传送门：

https://www.luogu.org/problemnew/show/P4052

题意：

给出$n$个单词，要求这$n$个单词至少有一个出现在长度为$m$的文章中的方案数。

思路：

如果改成没有出现过的呢？
首先可以发现答案就是所有串的个数减去不包含可读串的串的个数。
前半部分是$26^m$。后半部分$\text{dp}$求解。
$d{p}_{i,j}$表示匹配到第$i$个单词，在自动机第$j$个位置失配的方案数。
若当前不是$\text{end}$，则可以更新状态。
最后匹配完$m$位即可。

代码：

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#define mod 10007
using namespace std;
    int n,m,len=0;
    char s[10010];
    int dp[110][10100];
queue<int> f;
struct node
{
    int fail,tot;
    int son[30];
    node()
    {
        fail=tot=0;
        memset(son,0,sizeof(son));
    }
} tr[10010];
void build(char *s)
{
    int l=strlen(s),x=0;
    for(int i=0;i<l;i++)
    {
        if(!tr[x].son[s[i]-'A']) tr[x].son[s[i]-'A']=++len;
        x=tr[x].son[s[i]-'A'];
    }
    tr[x].tot=1;
}
void get_fail()
{
    while(!f.empty()) f.pop();
    for(int i=0;i<26;i++)
        if(tr[0].son[i])
        {
            tr[tr[0].son[i]].fail=0;
            f.push(tr[0].son[i]);
        }
    while(!f.empty())
    {
        int x=f.front();
        f.pop();
        for(int i=0;i<26;i++)
        {
            int y=tr[x].son[i];
            if(y)
            {
                tr[y].fail=tr[tr[x].fail].son[i];
                tr[y].tot|=tr[tr[y].fail].tot;
                f.push(y);
            }
            else tr[x].son[i]=tr[tr[x].fail].son[i];
        }
    }
}
int dg(int x,int k)
{
    if(!k) return 1;
    int op=dg(x,k>>1);
    if(k&1) return op*op%mod*x%mod; else return op*op%mod;
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        build(s);
    }
    tr[0].fail=0;
    get_fail();
    dp[0][0]=1;
    for(int i=1;i<=m;i++)
        for(int j=0;j<=len;j++)
            for(int k=0;k<26;k++)
                if(!tr[tr[j].son[k]].tot) dp[i][tr[j].son[k]]=(dp[i][tr[j].son[k]]+dp[i-1][j])%mod;
    int ans=dg(26,m);
    for(int i=0;i<=len;i++)
        ans=(ans-dp[m][i]+mod)%mod;
    printf("%d",ans);
}