【BZOJ 2553】[BeiJing2011]禁忌 AC自动机+期望概率dp

我一开始想的是倒着来，发现太屎，后来想到了一种神奇的方法——我们带着一个既有期望又有概率的矩阵，偶数(2*id)代表期望，奇数(2*id+1)代表概率，初始答案矩阵一列，1的位置为1(起点为0)，工具矩阵上如果是直接转移那么就是由i到j概率期望都乘上1/alphabet，特别的，对于一个包含禁忌串的节点直接由其父节点指向0，而且在计算期望是多加上他的概率，最后统计答案时把答案矩阵上所有的期望加和即可，这个方法很完美的被卡精了.......

#include <cstdio>
#include <cstring>
#include <vector>
typedef long double LD;
const int N=;
char s[];
int alpha,n,m;
struct Trie{
  int ch[],fail,deep;
  bool god;
}node[N];
int sz,q[N],size;
LD a[N][N],temp_a[N][N],b[N],temp_b[N],need;
inline void insert(char *w){
  int p=;
  for(int i=;w[i];i++){
    if(node[p].ch[w[i]-'a']==)node[p].ch[w[i]-'a']=++sz,node[sz].deep=node[p].deep+;
    p=node[p].ch[w[i]-'a'];
  }
  node[p].god=;
}
inline void build(){
  q[]=;
  for(int i=,j=;i<=j;i++){
    for(int l=;l<alpha;l++)
      if(node[q[i]].ch[l])
        q[++j]=node[q[i]].ch[l],
        node[q[j]].fail=q[i]?node[node[q[i]].fail].ch[l]:,
        node[q[j]].god=node[q[j]].god||node[node[q[j]].fail].god;
      else
        node[q[i]].ch[l]=q[i]?node[node[q[i]].fail].ch[l]:;
  }
}
void dfs(int x){
    for(int i=;i<alpha;i++){
        if(node[node[x].ch[i]].god){
            a[][x<<]+=need;
            a[][(x<<)+]+=need;
            a[][(x<<)+]+=need;
            continue;
        }
        a[node[x].ch[i]<<][x<<]+=need;
        a[(node[x].ch[i]<<)+][(x<<)+]+=need;
        if(node[node[x].ch[i]].deep>node[x].deep)
            dfs(node[x].ch[i]);
    }
}
inline void Multi_a(){
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      temp_a[i][j]=;
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      for(int k=;k<=size;k++)
        temp_a[i][j]+=a[i][k]*a[k][j];
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      a[i][j]=temp_a[i][j];
}
inline void Multi_b(){
  for(int i=;i<=size;i++)temp_b[i]=;
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      temp_b[i]+=a[i][j]*b[j];
  for(int i=;i<=size;i++)b[i]=temp_b[i];
}
int main(){
  scanf("%d%d%d",&n,&m,&alpha),need=./alpha;
  for(int i=;i<=n;i++)
    scanf("%s",s),insert(s);
  build(),dfs(),size=(sz<<)+,b[]=.;
  while(m){
    if(m&)Multi_b();
    m>>=,Multi_a();
  }
  LD ans=.;
  for(int i=;i<=sz;i++)ans+=b[i<<];
  printf("%lf",(double)ans);
  return ;
}

只是错了最后一个点

然后我去接受正解，按照我们的思路我们会建出来一个概率矩阵a[i][j]表示由j转移到i的概率，那就是加上1/alphabet，当然这里也有从禁忌串直接调回0的操作，然后我们想记录答案，我们就把矩阵扩大一阶，标号为sz+1，就是记录答案，那么我们由禁忌串向sz+1加上1/alphabet，那么我们发现这个矩阵自乘多少次，他的a[sz+1][0]的答案就是我们想要的答案。

经验教训：对于矩阵乘法，答案矩阵也可以是方阵，而且在一个矩阵里意义可以不同。

#include <cstdio>
#include <cstring>
#include <vector>
typedef long double LD;
const int N=;
char s[];
int alpha,n,m;
struct Trie{
  int ch[],fail,deep;
  bool god;
}node[N];
int sz,q[N],size;
LD a[N][N],temp[N][N],b[N][N],need;
inline void insert(char *w){
  int p=;
  for(int i=;w[i];i++){
    if(node[p].ch[w[i]-'a']==)node[p].ch[w[i]-'a']=++sz,node[sz].deep=node[p].deep+;
    p=node[p].ch[w[i]-'a'];
  }
  node[p].god=;
}
inline void build(){
  q[]=;
  for(int i=,j=;i<=j;i++){
    for(int l=;l<alpha;l++)
      if(node[q[i]].ch[l])
        q[++j]=node[q[i]].ch[l],
        node[q[j]].fail=q[i]?node[node[q[i]].fail].ch[l]:,
        node[q[j]].god=node[q[j]].god||node[node[q[j]].fail].god;
      else
        node[q[i]].ch[l]=q[i]?node[node[q[i]].fail].ch[l]:;
  }
}
void dfs(int x){
    for(int i=;i<alpha;i++){
        if(node[node[x].ch[i]].god){
            a[][x]+=need;
            a[sz+][x]+=need;
            continue;
        }
        a[node[x].ch[i]][x]+=need;
        if(node[node[x].ch[i]].deep>node[x].deep)
            dfs(node[x].ch[i]);
    }
}
inline void Multi_a(){
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      temp[i][j]=;
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      for(int k=;k<=size;k++)
        temp[i][j]+=a[i][k]*a[k][j];
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      a[i][j]=temp[i][j];
}
inline void Multi_b(){
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      temp[i][j]=;
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      for(int k=;k<=size;k++)
        temp[i][j]+=b[i][k]*a[k][j];
  for(int i=;i<=size;i++)
    for(int j=;j<=size;j++)
      b[i][j]=temp[i][j];
}
int main(){
  scanf("%d%d%d",&n,&m,&alpha),need=(LD)./(LD)alpha;
  for(int i=;i<=n;i++)
    scanf("%s",s),insert(s);
  build(),dfs(),size=sz+,a[size][size]=.;
  for(int i=;i<=size;i++)b[i][i]=.;
  while(m){
    if(m&)Multi_b();
    m>>=,Multi_a();
  }
  printf("%.6lf",(double)b[size][]);
  return ;
}

AC code