BZOJ1030——文本生成器

给你若干给字符串,再给你一个m,问长度是m的字符串中包含给定字符串的数量mod 10007是多少


这个拿过来啥思路也没有,后来还是看了题解,才知道,原来,原来。。。。那个带fail的Trie还可以搞别的

网上的大多数都是26^m-补集做的,麻烦啊,网上有一个非洲猴的blog写的挺好的
设f[i][j][k]表示什么呢,表示i:0..1 j:0..m k:0...size
表示i状态,匹配到j,在Trie上是k的数量
f[i | val[next[c]]][j + 1][next[c]] += f[i][j][k];对就是这样
我已经把next[c] == NULL的都变成fail了,
提醒按照lrj的白书学的AC自动机的同学们,那个last数组在匹配的所有时候都要循环枚举

提醒我自己当k等于0是不是这个位置是无,而是一个未匹配的点,因为计算last和dp时用last耗时代码量费啊


#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
struct AC{
    int ch[100010][30];
    int val[100010];
    char s[100010];
    int f[100010];
    int last[100010];
    int g[2][200][10010];
    int size;
    int len;
    inline int num(char x){
        return x - 'A';
    }
    inline void init(){
        memset(ch, 0, sizeof(ch));
        memset(val, 0, sizeof(val));
        size = 0;
        return;
    }
    inline void insert(){
        int u = 0;
        for(int i = 0; i < len; i ++){
            int c = num(s[i]);
            if(!ch[u][c]){
                ch[u][c] = ++ size;
            }
            u = ch[u][c];
        } 
        val[u] = 1;
        return;
    }
    inline void getfail(){
        queue<int> Q;
        for(int i = 0; i < 26; i ++) {
            if(ch[0][i]) {
                Q.push(ch[0][i]);
            }
        }
        while(!Q.empty()){
            int r = Q.front(); Q.pop();
            for(int i = 0; i < 26; i ++){
                int u = ch[r][i];
                if(!u){
                    ch[r][i] = ch[f[r]][i];
                    continue;
                }
                Q.push(u);
                int v = f[r];
                while(v && !ch[v][i]) v = f[v];
                f[u] = ch[v][i];
            }
        }
        return;
    }
    inline int dp(int m){
        for(int i = 0; i <= size; i ++){
            if(!val[i]){
                int j = f[i];
                while(j){
                    if(val[j] == 1){
                        val[i] = 1;
                        break;
                    }
                    j = f[j];
                }
            }
        }
        g[0][0][0] = 1;
        for(int i = 0; i <= 1; i ++){
            for(int j = 0; j < m; j ++){
                for(int k = 0; k <= size; k ++){
                    for(int c = 0; c < 26; c ++){
                        g[i | val[ch[k][c]]][j + 1][ch[k][c]] += g[i][j][k];
                        g[i | val[ch[k][c]]][j + 1][ch[k][c]] %= 10007;
                    }
                }
            }
        }
        int ret = 0;
        for(int k = 0; k <= size; k ++) {
            ret += g[1][m][k];
            ret %= 10007;
        }
        return ret;
    }
} wt;
int main(){
    int N, M;
    scanf("%d%d", &N, &M);
    wt.init();
    for(int i = 1; i <= N; i ++){
        scanf("%s", wt.s);
        wt.len = strlen(wt.s);
        //if(wt.len > M) continue;
        wt.insert();
    }
    wt.getfail();
    printf("%d\n", wt.dp(M));
    return 0;
}


你可能感兴趣的:(C++,AC自动机,1030,bzoj)