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 = 22;
const int mod = 1e9+7;
const int maxnode = 20*20+10;
const int sigma_size = 62;
struct AhoCorasickAutomata
{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int sz;
    int f[maxnode];
    int last[maxnode];
    void clear(){sz = 1; memset(ch[0],0,sizeof ch[0]); }
    int idx(char c){
        if(c>='a'&&c<='z') return c-'a';
        else if(c>='A'&&c<='Z') return c-'A'+26;
        else return c-'0'+52;
    }

    void insert(char *s,int v)
    {
        int u = 0, n = strlen(s);
        for(int i = 0; i < n; i++){
            int c = idx(s[i]);
            if(!ch[u][c]){
                memset(ch[sz], 0, sizeof ch[sz]);
                val[sz] = 0;
                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[0] = 0;
        for(int c = 0; c < sigma_size; c++){
            int u = ch[0][c];
            if(u){f[u] = 0; q.push(u); last[u] = 0;}
        }

        while(!q.empty()){
            int r = q.front(); q.pop();
            for(int c = 0; 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[22];
char a[66];
double p[66];
double dp[102][200*200+10];
const double eps = 1e-8;
int L, n;
double dfs(int i,int j)///第一維表示第i個位置選擇字符。j表示前面的字符串匹配後在自動機上的節點編號。
{
    if(ac.val[j]||ac.last[j]) return 1;///該位置在某個模板串的結尾。
    if(i==L) return 0;
    if(dp[i][j]>-eps) return dp[i][j];
    double &res = dp[i][j];
    res = 0;
    for(int k = 0; k < n; k++){
        int x = ac.find(j,a[k]);
        res += p[k]*dfs(i+1,x);
    }
    return res;
}
int main()
{
    int T, k;
    int cas = 1;
    cin>>T;
    while(T--)
    {
        scanf("%d",&k);
        ac.clear();
        for(int i = 1; i <= k; i++){
            scanf("%s",s);
            ac.insert(s,1);
        }
        ac.getFail();
        scanf("%d",&n);
        for(int i = 0; i < n; i++){
            cin>>a[i]>>p[i];
        }
        scanf("%d",&L);
        ms(dp,-1);
        printf("Case #%d: %f\n",cas++,1-dfs(0,0));
    }
    return 0;
}