UVA11468 Substring --- AC自动机

题解： 按照蓝书的解法，先把所有单词建trie树，然后生成一个字符就等价于在trie上面走一步，只要不走到单词节点即可，所有用dp(u,l)表示当前在u节点上，还剩l个字符时，满足题意的概率。

需要注意的是，getFail()中if(isWord[]) ... 的那行代码比较关键，缺掉就会WA。原因是在常规的AC自动机中，在字符串中每移动一个字符，就要反复通过fail指针来找是否有以当前字符结尾的匹配串。但是在本题的代码中，每走一步就直接通过isWord判断是否是匹配串结尾，就缺少了fail指针反复找匹配串的那一步，所以需要在预处理fail的时候提前把它处理好。比如模式串有abcd,bc,生成串是abc,在走到c点的时候，就需要不通过跳fail指针，直接判断当前遇到匹配串了，这么说可能没说清楚，具体见代码

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;

char patterns[30][25];
int k,n,L;
double prob[63];
const int maxn = 450;
int ch[maxn][63];
int f[maxn];
bool isWord[maxn];
int cnt;

int idx(char c) {
    if(c>='a'&&c<='z') return c-'a';
    else if(c>='A'&&c<='Z') return 26+c-'A';
    else return 52+c-'0';
}

void insert(char *s) {
    int len = strlen(s);
    int now = 0;
    for(int i = 0;i < len;i++) {
        int index = idx(s[i]);
        if(ch[now][index] == 0) {
            ch[now][index] = cnt++;
            now = ch[now][index];
        } else {
            now = ch[now][index];
        }
    }
    isWord[now] = true;
}

void getFail() {
    f[0] = 0;
    queue<int> q;    
    for(int i = 0;i < 63;i++) {
        if(ch[0][i]) {
            q.push(ch[0][i]);
            f[ch[0][i]] = 0;
        }
    }
    while(!q.empty()) {
        int now = q.front(); q.pop();
        for(int i = 0;i < 63;i++) {
            if(ch[now][i]) {
                f[ch[now][i]] = ch[f[now]][i];
                q.push(ch[now][i]);
                if(isWord[ch[f[now]][i]]) isWord[ch[now][i]]=true;
            } else {
                ch[now][i] = ch[f[now]][i];
            }
        }
    }
}

double dp[maxn][110];
bool vis[maxn][110];
// baaaaaaaa
double DP(int u,int l) {
    if(l==0) return 1.0;
    if(vis[u][l]) return dp[u][l];
    vis[u][l] = true;
    double &ans = dp[u][l];
    ans = 0;
    for(int i = 0;i < 63;i++) {
        if(!isWord[ch[u][i]]) {
            ans += prob[i]*DP(ch[u][i],l-1);
        }
    }
    return ans;
}

int main() {
    int T;
    //freopen("out.txt","w",stdout);
    scanf("%d",&T);
    int kase = 1;
    while(T--) {
        memset(vis,0,sizeof(vis));
        cnt = 1;
        scanf("%d",&k);
        for(int i = 0;i < k ;i++) scanf("%s",patterns[i]);
        scanf("%d",&n);
        for(int i = 0;i < 63;i++) prob[i] = 0;
        for(int i = 0;i < n;i++) {
            char s[5];
            double t;
            scanf("%s%lf",s,&t);
            prob[idx(s[0])] = t;
        }
        scanf("%d",&L);
        memset(isWord,0,sizeof(isWord));
        memset(ch,0,sizeof(ch));
        memset(f,0,sizeof(f));
        memset(dp,0,sizeof(dp));
        for(int i = 0;i < k;i++) insert(patterns[i]);
        getFail();
        printf("Case #%d: %.6lf\n",kase++,DP(0,L));
    }

    return 0;
}