BZOJ 1030: [JSOI2007]文本生成器 AC自动机

1030: [JSOI2007]文本生成器

Description

　　JSOI交给队员ZYX一个任务，编制一个称之为“文本生成器”的电脑软件：该软件的使用者是一些低幼人群，
他们现在使用的是GW文本生成器v6版。该软件可以随机生成一些文章―――总是生成一篇长度固定且完全随机的文
章—— 也就是说，生成的文章中每个字节都是完全随机的。如果一篇文章中至少包含使用者们了解的一个单词，
那么我们说这篇文章是可读的（我们称文章a包含单词b，当且仅当单词b是文章a的子串）。但是，即使按照这样的
标准，使用者现在使用的GW文本生成器v6版所生成的文章也是几乎完全不可读的?。ZYX需要指出GW文本生成器 v6
生成的所有文本中可读文本的数量，以便能够成功获得v7更新版。你能帮助他吗？

Input

　　输入文件的第一行包含两个正整数，分别是使用者了解的单词总数N (<= 60)，GW文本生成器 v6生成的文本固
定长度M；以下N行，每一行包含一个使用者了解的单词。这里所有单词及文本的长度不会超过100，并且只可能包
含英文大写字母A..Z

Output

　　一个整数，表示可能的文章总数。只需要知道结果模10007的值。

Sample Input

2 2
A
B

Sample Output

100

思路：

　　建Trie图跑DP， f[i][j]表示到了i这个点已经跑了j步的不可行方案数，当且仅当一个点及它要更新的儿子都不是危险结点的时候才可以更新。再用26^n-跑了m步的方案数就好辣！

#include <cstdio>
#include <iostream>
#include <queue>
#include <cstring>
using namespace std;
const int N = 21000;
const int mod = 10007;
int ch[N][27], fail[N], n, m, cnt=1;
char s[N];
bool flg[N];
int f[200][N];
void Insert() {
    int p = 1;
    int len = strlen(s);
    for(int i=0;i<len;i++) {
        int &k = ch[p][s[i]-'A'+1];
        if(!k)k=++cnt;
        p = k;
    }
    flg[p]=1;
}
void Get_Fail() {
    int p;
    for(int i=1;i<=26;i++) ch[0][i] = 1;
    queue<int>q;
    q.push(1);
    while(!q.empty()) {
        p = q.front();q.pop();
        for(int i=1;i<=26;i++) {
            if(ch[p][i]) fail[ch[p][i]] = ch[fail[p]][i], q.push(ch[p][i]);
            else ch[p][i] = ch[fail[p]][i];
        }
        flg[p]|=flg[fail[p]];
    }
}
int main() {
    scanf("%d%d", &n, &m);
    for(int i=1;i<=n;i++) {
        scanf("%s", s);
        Insert();
    }
    Get_Fail();
    f[0][1] = 1;
    for(int i=1;i<=m;i++) {
        for(int p=1;p<=cnt;p++) {
            if((!flg[p])&&f[i-1][p]) {
                for(int k=1;k<=26;k++) {
                    (f[i][ch[p][k]] += f[i-1][p])%=mod;
                }
            }
        }
    }
    int ans =1;
    for(int i=1;i<=m;i++) (ans*=26)%=mod;
    for(int i=1;i<=cnt;i++) {
        if(!flg[i])((ans-=f[m][i])+=mod)%=mod;
    }
    printf("%d
", ans);
}

View Code