【uva11468-Substring】AC自动机+dp

http://acm.hust.edu.cn/vjudge/problem/31655

题意：给定k个模板串，n个字符以及选择它的概率pro[i]，要构造一个长度问L的字符串s，问s不包含任意一个模板串的概率。

题解：

ed[i]标记trie上的点i是不是任意一个模板串的结尾（在求fail的时候ed[i]|=ed[i.fail]）
d[i][l]表示从i出发还要走l步，构造的串不含模板串的概率，dp一下。注意清零。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<queue>
 using namespace std;

 const int N=,L=,S=;
 char s[L];
 int cnt[N*L];
 double d[N*L][L],pro[];
 bool ed[N*L];
 queue<int> q;
 int num,k,n,l;
 struct node{
     int son[];
     int fail;
 }a[N*L];

 int idx(char c)
 {
     if(c<='z' && c>='a') return c-'a'+;
     if(c<='Z' && c>='A') return c-'A'+;
     return c-''+;
 }

 void clear(int x)
 {
     a[x].fail=;
     memset(a[x].son,,sizeof(a[x].son));
 }

 void trie(char *c)
 {
     int l=strlen(c);
     int x=;
     for(int i=;i<l;i++)
     {
         int t=idx(c[i]);
         if(!a[x].son[t])
         {
             num++;
             clear(num);
             a[x].son[t]=num;
         }
         x=a[x].son[t];
     }
     ed[x]=;
 }

 void buildAC()
 {
     while(!q.empty()) q.pop();
     for(int i=;i<=S;i++)
         if(a[].son[i]) q.push(a[].son[i]);
     while(!q.empty())
     {
         int x=q.front();q.pop();
         int fail=a[x].fail;
         for(int i=;i<=S;i++)
         {
             int y=a[x].son[i];
             if(y)
             {
                 a[y].fail=a[fail].son[i];
                 ed[y]|=ed[a[fail].son[i]];
                 q.push(y);
             }
             else a[x].son[i]=a[fail].son[i];
         }
     }
 }

 double count(int x,int l)
 {
     if(l==) return d[x][l]=1.0;
     if(d[x][l]!=-) return d[x][l];
     d[x][l]=;
     for(int i=;i<=S;i++)
     {
         if(!pro[i]) continue;
         int y=a[x].son[i];
         if(!ed[y]) d[x][l]+=pro[i]*count(y,l-);
     }
     return d[x][l];
 }

 int main()
 {
     freopen("a.in","r",stdin);
     freopen("a.out","w",stdout);
     int T,cas=;
     scanf("%d",&T);
     while(T--)
     {
         num=;
         clear();
         memset(cnt,,sizeof(cnt));
         memset(ed,,sizeof(ed));
         memset(pro,,sizeof(pro));
         scanf("%d",&k);
         for(int i=;i<=k;i++)
         {
             scanf("%s",s);
             trie(s);
         }
         buildAC();
         scanf("%d",&n);getchar();
         for(int i=;i<=n;i++)
         {
             char c;
             scanf("%c",&c);getchar();
             scanf("%lf",&pro[idx(c)]);getchar();
         }
         scanf("%d",&l);
         for(int i=;i<=num;i++)
             for(int j=;j<=l;j++)
                 d[i][j]=-;
         printf("Case #%d: %lf\n",++cas,count(,l));
     }
     return ;
 }