[BZOJ]1030 文本生成器

[BZOJ]1030

题意：

　　给出一个$M$，表示有$M$个字母是随机给出的，有很多种可能。再给出$N$个字符串，求$N$在$M$中出现的可能个数。

题解：

　　类似于[BZOJ]1009。未知字符求个数一般都是字符串Dp，解决方法类似。不过这个题要先转化为不出现的可能个数再拿总数减去。
　　这个题是多个串匹配，所以是$AC自动机$，用$dp[i][j]$表示到了第$i$个字符($M$中)，匹配到了第$j$个节点的方案数。转移的时候要注意不能转移到单词节点(即不能完全匹配)，最后的答案就是$26^m - \sum_{k=0}^{siz}dp[m][k]$。
　　首先AC自动机我不会，所以还是背一波板子233。

代码

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>

#define mod 10007

using namespace std;

const int Maxn = 5010;

queue <int> Q;

int n,m,tot,f[Maxn],dp[105][Maxn],ch[Maxn][28],vis[Maxn];
char s[105];

int Pow(int a,int n){int c = 1;
    while(n){if(n & 1){c = (c * a) % mod;}a = (a * a) % mod;n >>= 1;}
    return c % mod;
}

void Insert(){
    int len = strlen(s),u = 0;
    for(int i = 0;i < len;i++){int v = s[i] - 64;
        if(ch[u][v] == 0)ch[u][v] = ++tot;
        u = ch[u][v];
    }vis[u] = 1;
}

void Build(){
    for(int i = 1;i <= 26;i++)if(ch[0][i] > 0)Q.push(ch[0][i]);
    while(!Q.empty()){
        int u = Q.front();Q.pop();
        vis[u] |= vis[f[u]];
        for(int i = 1;i <= 26;i++){
            int v = ch[u][i];
            if(v > 0){
                f[v] = ch[f[u]][i];
                Q.push(v);
            }
            else ch[u][i] = ch[f[u]][i];
        }
    }
}

int main(){
//  freopen("textgen.in","r",stdin);
//  freopen("textgen.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i++){scanf("%s",s);Insert();}
    Build();dp[0][0] = 1;
    for(int i = 1;i <= m;i++){
        for(int j = 0;j <= tot;j++){
            if(vis[j] == 0 && dp[i - 1][j] > 0){
                for(int k = 1;k <= 26;k++){
                    dp[i][ch[j][k]] = (dp[i][ch[j][k]] + dp[i - 1][j]) % mod;
                }
            }
        }
    }int Ans1 = 0,Ans2 = Pow(26,m);
    for(int i = 0;i <= tot;i++)if(vis[i] == 0)Ans1 += dp[m][i];
    printf("%d\n",((Ans2 - Ans1) % mod + mod) % mod);
    return 0;
}