uva11468 AC自动机&&树形DP

题意：给定一些字符和选择概率，随机选择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;
}