UVA 11468 AC 自动机

首先我们应该是枚举 L个位置上的每个字符来得到最终概率

然后AC自动机的作用就是为了判断你枚举的地方是否对应了单词节点，如果对应了，就肯定要不得

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int max_node = ;
const int N =;
const int node_num=;
int idx[N],k,n;
char str[][];
double prob[]; //这个地方一开始开小了，题目也没说n为多少。坑啊。。一直RE没找到原因
int vis[max_node][];
double d[max_node][];
struct ACtrie
{
    int ch[max_node][];
    int sz;
    int val[max_node];
    int fail[max_node];
    //int last[max_node];
    void init(){
        memset(ch,,sizeof ch);
        memset(val,,sizeof val);
        memset(fail,,sizeof fail);
        //memset(last,0,sizeof last);
        sz=;
    }
    void insert(char* s,int v){
        int u=;
        for (int i=;s[i];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]=;
    }
    void getfail(){
        queue<int> q;
        fail[]=;
        for (int i=;i<node_num;i++)
            if (ch[][i]){
                fail[ch[][i]]=;
                q.push(ch[][i]);
            }

        while (!q.empty()){
            int u=q.front();
            q.pop();
            for (int c=;c<node_num;c++){
                int v=ch[u][c];
                if (!v) {ch[u][c] = ch[fail[u]][c];continue;} //这个个普通AC自动机不一样，因为要一视同仁所有字符，当前字符在该子树如果找不到就可以转到其失败节点去，因为失败节点有的话，同样是满足条件的
                q.push(v);
                int p=fail[u];
                while (p && !ch[p][c]) p=fail[p];
                fail[v]=ch[p][c];
                //last[v]=val[fail[p]]?fail[p] : last[fail[p]];
                val[v]|=val[fail[v]];
            }
        }
    }
} T;
double getprob(int u,int L)
{
    if (L==) return 1.0;
    if (vis[u][L]) return d[u][L];
    vis[u][L]=;
    d[u][L]=0.0;
    for (int i=;i<n;i++){
        if (!T.val[T.ch[u][i]]) d[u][L]+=prob[i]*getprob(T.ch[u][i],L-);
    }
    return d[u][L];
}
int main()
{
    int t,L;
    char cc[];
    scanf("%d",&t);
    int kase=;
    while (t--){
        T.init();
        scanf("%d",&k);
        for (int i=;i<k;i++)
            scanf("%s",str[i]);
        scanf("%d",&n);
        for (int i=;i<n;i++){
          scanf("%s%lf",&cc,&prob[i]);
          idx[cc[]]=i;
        }
        for (int i=;i<k;i++){
            T.insert(str[i],);
        }
        T.getfail();
        memset(vis,,sizeof vis);
        scanf("%d",&L);
        double ans=getprob(,L);
        printf("Case #%d: %.6lf\n",++kase,ans);
    }
    return ;
}