各种手残和手贱,导致写出来的都是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;
}