题解: 按照蓝书的解法,先把所有单词建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;
}