BZOJ 1030 TRIE图+DP

貌似是论问题？、、

感觉挺好玩了，写起来也简单。。

https://files.cnblogs.com/proverbs/Trie%E5%9B%BE%E7%9A%84%E6%9E%84%E5%BB%BA_%E6%B4%BB%E7%94%A8%E4%B8%8E%E6%94%B9%E8%BF%9B.pdf

#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>

#define N 111
#define MOD 10007

using namespace std;

struct TR
{
    int son[26]; bool fg; int f;
}tr[N*N];

int n,cnt,m,ans;
int q[N*N];
char str[N];
int dp[N][N*N][2];

inline void insert(char *s)
{
    int len=strlen(s+1);
    int now=1;
    for(int i=1;i<=len;i++)
    {
        if(!tr[now].son[s[i]-'A']) tr[now].son[s[i]-'A']=++cnt;
        now=tr[now].son[s[i]-'A'];
    }
    tr[now].fg=true;
}

inline void build()
{
    int h=1,t=1,sta; q[1]=1;
    while(h<=t)
    {
        sta=q[h++];
        for(int i=0;i<26;i++)
        {
            int now=tr[sta].son[i],tmp;
            if(sta==1) tmp=1;
            else tmp=tr[tr[sta].f].son[i];
            if(now==0) tr[sta].son[i]=tmp;
            else
            {
                tr[now].f=tmp;tr[now].fg|=tr[tmp].fg;
                q[++t]=now;
            }
        }
    }
}

inline void read()
{
    cnt=1; tr[1].f=1;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",str+1);
        insert(str);
    }
    build();
}

inline void go()
{
    dp[0][1][0]=1;
    for(int i=0;i<m;i++)
        for(int j=1;j<=cnt;j++)
        {
            if(dp[i][j][0])
            {
                for(int k=0;k<26;k++)
                {
                    if(tr[tr[j].son[k]].fg) dp[i+1][tr[j].son[k]][1]=(dp[i+1][tr[j].son[k]][1]+dp[i][j][0])%MOD;
                    else dp[i+1][tr[j].son[k]][0]=(dp[i+1][tr[j].son[k]][0]+dp[i][j][0])%MOD;
                }
            }
            if(dp[i][j][1])
            {
                for(int k=0;k<26;k++)
                    dp[i+1][tr[j].son[k]][1]=(dp[i+1][tr[j].son[k]][1]+dp[i][j][1])%MOD;
            }
        }
    for(int i=1;i<=cnt;i++) ans=(ans+dp[m][i][1])%MOD;
    printf("%d\n",ans);
}

int main()
{
    read(),go();
    return 0;
}