bzoj1030 [JSOI2007]文本生成器

Description

---

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

Solution

---

正难则反，我们可以统计一个都不包含的情况
那么就是建一棵trie，求有多少条路径刚好走了m步且不经过任意一个代表单词结尾的节点，dp一下就行了
注意数组的范围和开ll的问题

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int MOD=10007;
const int N=65;
const int L=105;

int rec[N*L][26],fail[N*L],tot;
int f[L][N*L],p[N*L];
int queue[N*L];

char s[L];

void insert(char *str) {
    int len=strlen(str+1);
    int now=0;
    rep(i,1,len) {
        int ch=str[i]-'A';
        if (!rec[now][ch]) rec[now][ch]=++tot;
        now=rec[now][ch];
    }
    p[now]++;
}

void get_fail() {
    int head=1,tail=0;
    rep(i,0,25) if (rec[0][i]) queue[++tail]=rec[0][i];
    while (head<=tail) {
        int now=queue[head++];
        rep(i,0,25) if (rec[now][i]) {
            int tmp=fail[now];
            while (tmp&&!rec[tmp][i]) tmp=fail[tmp];
            fail[rec[now][i]]=rec[tmp][i];
            if (p[rec[tmp][i]]) p[rec[now][i]]=1;
            // p[rec[now][i]]=p[rec[tmp][i]];
            queue[++tail]=rec[now][i];
        }
    }
}

void solve(int n,int m) {
    f[0][0]=1;
    rep(i,1,m) rep(j,0,tot) {
        if (!p[j]&&f[i-1][j]) {
            rep(k,0,25) {
                int tmp=j;
                while (tmp&&!rec[tmp][k]) tmp=fail[tmp];
                f[i][rec[tmp][k]]=(f[i][rec[tmp][k]]+f[i-1][j])%MOD;
            }
        }
    }
    LL ans1=1,ans2=0;
    rep(i,1,m) ans1=ans1*26%MOD;
    rep(i,0,tot) if (!p[i]) ans2=(ans2+f[m][i])%MOD;
    // printf("%lld %lld\n", ans2,ans1);
    printf("%lld\n", (ans1-ans2+MOD)%MOD);
}

int main(void) {
    // freopen("data.in","r",stdin);
    // freopen("myp.out","w",stdout);
    int n,m; scanf("%d%d",&n,&m);
    rep(i,1,n) {
        scanf("%s",s+1);
        insert(s);
    }
    get_fail();
    solve(n,m);
    return 0;
}