UVA 11468 —— Substring（AC自动机+DP）

各种手残和手贱，导致写出来的都是WA自动机，几番周折才AC的说。

训练指南里提到的例题，书上用记忆化搜索，我写成递推的。

dp[i][j]代表长度为i，并且匹配到j状态时的概率。

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
int t, n, m, l;
double dp[101][401], p[62];
char s[30], w[5];
struct ACAuto{
    int ch[401][62], f[401];
    bool val[401];
    int sz;
    void init(){
        sz=1;
        val[0]=0;
        memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c){
        if(c>='0' && c<='9')    return c-'0';
        if(c>='a' && c<='z')    return c-'a'+10;
        return c-'A'+36;
    }
    void insert(){
        int len=strlen(s);
        int u=0;
        for(int i=0; i<len; i++){
            int c=idx(s[i]);
            if(!ch[u][c]){
                val[sz]=0;
                memset(ch[sz],0,sizeof(ch[sz]));
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=1;
    }
    void getfail(){
        queue<int> Q;
        f[0]=0;
        for(int i=0; i<62; i++){
            int u=ch[0][i];
            if(u){
                f[u]=0; Q.push(u);
            }
        }
        while(!Q.empty()){
            int r=Q.front(); Q.pop();
            for(int i=0; i<62; 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];
                val[u]|=val[f[u]];
            }
        }
    }
    double solve(){
        memset(dp,0.0,sizeof(dp));
        dp[0][0]=1.0;
        for(int i=0; i<l; i++){
            for(int j=0; j<sz; j++){
                if(val[j])  continue;
                for(int k=0; k<62; k++){
                    int u=ch[j][k];
                    if(val[u])  continue;
                    dp[i+1][u]+=dp[i][j]*p[k];
                }
            }
        }
        double ans=0.0;
        for(int i=0; i<sz; i++) ans+=dp[l][i];
        return ans;
    }
}ac;
int main(){
    scanf("%d", &t);
    for(int ct=1; ct<=t; ct++){
        scanf("%d", &n);
        ac.init();
        while(n--){
            scanf("%s", s);
            ac.insert();
        }
        scanf("%d", &m);
        memset(p,0.0,sizeof(p));
        double tmp;
        for(int i=0; i<m; i++){
            scanf("%s %lf", w, &tmp);
            p[ac.idx(w[0])]+=tmp;
        }
        ac.getfail();
        scanf("%d", &l);
        printf("Case #%d: %.6lf\n", ct, ac.solve());
    }
    return 0;
}