题意:给定一些字符和选择概率,随机选择L此得到长度为L的随机字符串。再给出K个模板串,计算S不包含任何一个模板串的概率。
将所有可能字符加入Trie树,建立AC自动机,定义match数组。
match[u]=1:u字符串含有模板子串。如何计算match数组?
初始化:match[u]=1:u是模板串结尾结点
递推公式:match[u]|=match[f[u]](u本身是模板串/u沿着失配边走到的节点字符串的子串是模板串)
从而题目转换为一个树上的问题:从根节点出发,至多走L步,每一步选择都有一个对应概率,求没有经过标记点的概率,可用树形DP解决。
pro[u][L]:在u节点处至多还要走L步没有走到标记点的概率。pro[u][L]=sigma(prob[u][v]*pro[v][L-1]),v是u的子节点,prob[u][v]是选择边<u,v>的概率(全概率公式)
总结:
1.将模板串和文本串中所有可能出现的字符加入Trie树(用idx数组将所有可能的字符对应成不同整数---将idx函数改成预处理idx数组,先建树,再插入),建立AC自动机,并不是仅仅将模板字符串插入Trie树(边建树,边插入)
2.预处理match数组:字符串u是否含有模板子串!(利用fail数组的性质进行转移)
3.求概率:多用dp尝试;dp解决问题类型:每次有若干选择,一共选择若干次。
#include<cstdio>
#include<cstring>
#include<queue>
#include<map>
using namespace std;
const int SIGMA_SIZE=64;
const int MAXNODE=500;//节点个数
const int MAXS=20+10;//模板个数
int idx[256],n;
double prob[SIGMA_SIZE];
struct AC{
int ch[MAXNODE][SIGMA_SIZE];
int f[MAXNODE];//fail函数
int match[MAXNODE];//是否包含某个字符串
int sz;//节点总数
void init(){
sz=1;
memset(ch[0],0,sizeof(ch[0]));
}
//插入字符串
void insert(char *s){
int u=0,n=strlen(s);
for(int i=0;i<n;i++){
int c=idx[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]=1;//字符串结尾节点
}
//计算fail函数
void getfail(){
queue<int> q;
f[0]=0;
//初始化队列
for(int c=0;c<SIGMA_SIZE;c++){
int u=ch[0][c];
if(u) { f[u]=0;q.push(u); }
}
//按BFS顺序计算fail
while(!q.empty()){
int r=q.front();q.pop();
for(int c=0;c<SIGMA_SIZE;c++){
int u=ch[r][c];
if(!u) { ch[r][c]=ch[f[r]][c];continue; }
q.push(u);
int v=f[r];
while(v&&!ch[v][c]) v=f[v];
f[u]=ch[v][c];
match[u]|=match[f[u]];//u/f[u]是某串结尾节点=>u点都要被标记!
}
}
}
void dump(){
printf("sz=%d\n",sz);
for(int i=0;i,sz;i++)
printf("%d: %d %d %d\n",i,ch[i][0],ch[i][1],match[i]);
printf("\n");
}
};
AC ac;
double d[MAXNODE][105];
int vis[MAXNODE][105];
double getProb(int u,int L){
if(!L) return 1.0;
if(vis[u][L]) return d[u][L];
vis[u][L]=1;
double &ans=d[u][L];
ans=0.0;
for(int i=0;i<n;i++)
if(!ac.match[ac.ch[u][i]]) ans+=prob[i]*getProb(ac.ch[u][i],L-1);
return ans;
}
char s[30][30];
int main(){
int T;
//freopen("a.txt","r",stdin);
scanf("%d",&T);
for(int kase=1;kase<=T;kase++){
int k,L;
scanf("%d",&k);
for(int i=0;i<k;i++) scanf("%s",s[i]);
scanf("%d",&n);
for(int i=0;i<n;i++){
char ch[9];
scanf("%s%lf",ch,&prob[i]);
idx[ch[0]]=i;//按照可能出现的字符顺序依次标记
}
ac.init();
for(int i=0;i<k;i++) ac.insert(s[i]);
ac.getfail();
scanf("%d",&L);
memset(vis,0,sizeof(vis));
printf("Case #%d: %.6lf\n",kase,getProb(0,L));
}
return 0;
}