UVA 11468 AC自动机入门题 记忆化概率dp+ac自动机

/**
链接：https://vjudge.net/problem/UVA-11468
详见lrj训练指南P218

我的是反向求存在模板串的概率。
dp[i][j]表示当前i位置选择字符，前面i-1个字符在自动机的匹配节点编号为j时候的状态可以获得的存在概率。

书上的好简洁，求idx(c)直接利用已经给定的n个字符的下标作为结果。
正向求解！厉害。
*/
#include<bits/stdc++.h>
using namespace std;
#define P pair<int,int>
#define ms(x,y) memset(x,y,sizeof x)
#define LL long long
const int maxn = ;
const int mod = 1e9+;
const int maxnode = *+;
const int sigma_size = ;
struct AhoCorasickAutomata
{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int sz;
    int f[maxnode];
    int last[maxnode];
    void clear(){sz = ; memset(ch[],,sizeof ch[]); }
    int idx(char c){
        if(c>='a'&&c<='z') return c-'a';
        else if(c>='A'&&c<='Z') return c-'A'+;
        else return c-''+;
    }

    void insert(char *s,int v)
    {
        int u = , n = strlen(s);
        for(int i = ; i < n; i++){
            int c = idx(s[i]);
            if(!ch[u][c]){
                memset(ch[sz], , sizeof ch[sz]);
                val[sz] = ;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
    }

    int find(int j,char b){
        int c = idx(b);
        j = ch[j][c];
        //if(val[j]) print(j);
        //else if(last[j]) print(last[j]);
        return j;
    }

    void print(int j)
    {
        if(j){
            //cnt[val[j]]++;
            print(last[j]);
        }
    }

    void getFail(){
        queue<int> q;
        f[] = ;
        for(int c = ; c < sigma_size; c++){
            int u = ch[][c];
            if(u){f[u] = ; q.push(u); last[u] = ;}
        }

        while(!q.empty()){
            int r = q.front(); q.pop();
            for(int c = ; c < sigma_size; c++){
                int u = ch[r][c];
                if(!u){
                    ch[r][c] = ch[f[r]][c]; continue;
                }//if(!u) continue;
                q.push(u);
                int v = f[r];
                while(v&&!ch[v][c]) v = f[v];
                f[u] = ch[v][c];
                last[u] = val[f[u]] ? f[u] : last[f[u]];
            }
        }
    }

} ac ;
char s[];
char a[];
double p[];
double dp[][*+];
const double eps = 1e-;
int L, n;
double dfs(int i,int j)///第一维表示第i个位置选择字符。j表示前面的字符串匹配后在自动机上的节点编号。
{
    if(ac.val[j]||ac.last[j]) return ;///该位置在某个模板串的结尾。
    if(i==L) return ;
    if(dp[i][j]>-eps) return dp[i][j];
    double &res = dp[i][j];
    res = ;
    for(int k = ; k < n; k++){
        int x = ac.find(j,a[k]);
        res += p[k]*dfs(i+,x);
    }
    return res;
}
int main()
{
    int T, k;
    int cas = ;
    cin>>T;
    while(T--)
    {
        scanf("%d",&k);
        ac.clear();
        for(int i = ; i <= k; i++){
            scanf("%s",s);
            ac.insert(s,);
        }
        ac.getFail();
        scanf("%d",&n);
        for(int i = ; i < n; i++){
            cin>>a[i]>>p[i];
        }
        scanf("%d",&L);
        ms(dp,-);
        printf("Case #%d: %f\n",cas++,-dfs(,));
    }
    return ;
}