题意:给出一些字符和各自对应的选择概率,随机选择 L 次后将得到一个长度为 L 的随机字符串 S 。给出 K 个模板,计算 S 不包含任何一个串的概率。
思路:首先将模板字符串建成AC自动机,有单词的节点标记为禁止,然后再把 S 和模板串匹配,并求 S 匹配完且不走禁止点的概率。设d[ i ][ j ]为当前在 i 节点还剩下 j 步没匹配完的概率,那么很明显d[ i ][ j ]+=p[ k ]*d[ k ][ j-1 ],k为 i 节点的子节点且k不是禁止节点。
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=405;
const int size=65;
double p[size],d[maxn][maxn];
vector<int>V;
int vis[maxn][maxn];
struct AC_auto
{
int ch[maxn][size];
int match[maxn];
int f[maxn];
int sz;
void init()
{
sz=1;
memset(ch[0],0,sizeof(ch[0]));
memset(f,0,sizeof(f));
}
int index(char c)
{
if(c>='a'&&c<='z')
return c-'a';
if(c>='A'&&c<='Z')
return c-'A'+26;
return c-'0'+52;
}
void insert(char *s,int k)
{
int u,n=strlen(s);
u=0;
for(int i=0;i<n;i++)
{
int c=index(s[i]);
if(!ch[u][c])
{
memset(ch[sz],0,sizeof(ch[sz]));
match[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
}
match[u]=k;
}
void get_fail()
{
queue<int>q;
int u=0;
for(int i=0;i<62;i++)
if(ch[u][i])
{
q.push(ch[u][i]);
f[ch[u][i]]=0;
}
while(!q.empty())
{
u=q.front();
q.pop();
for(int i=0;i<62;i++)
if(ch[u][i])
{
int r=ch[u][i];
q.push(r);
int p=f[u];
while(p&&!ch[p][i])
p=f[p];
f[r]=ch[p][i];
match[r]|=match[f[r]];
}else
ch[u][i]=ch[f[u]][i];//只有match数组还不行,还要这一步辅助match数组
}
}
double dfs(int u,int len)
{
if(!len)
return 1.0;
if(vis[u][len])
return d[u][len];
vis[u][len]=1;
d[u][len]=0.0;
for(int i=0;i<V.size();i++)
{
int t=ch[u][V[i]];
if(!match[t])
d[u][len]+=p[V[i]]*dfs(t,len-1);
}
return d[u][len];
}
}ac;
int main()
{
int T,k,n,kase=0;
scanf("%d",&T);
while(T--)
{
int i,j;
char s[25];
ac.init();
scanf("%d",&k);
for(i=1;i<=k;i++)
{
scanf("%s",s);
ac.insert(s,i);
}
ac.get_fail();
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%s",s);
int t=ac.index(s[0]);
scanf("%lf",&p[t]);
V.push_back(t);
}
scanf("%d",&n);
memset(vis,0,sizeof(vis));
printf("Case #%d: %.6lf\n",++kase,ac.dfs(0,n));
V.clear();
}
}