码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2017-09-15 22:30:45      阅读:135      评论:0      收藏:0      [点我收藏+]

标签:main   opened   技术   none   答案   int   也有   pen   har   

我一开始想的是倒着来,发现太屎,后来想到了一种神奇的方法——我们带着一个既有期望又有概率的矩阵,偶数(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=200;
char s[20];
int alpha,n,m;
struct Trie{
  int ch[26],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=0;
  for(int i=0;w[i];i++){
    if(node[p].ch[w[i]-a]==0)node[p].ch[w[i]-a]=++sz,node[sz].deep=node[p].deep+1;
    p=node[p].ch[w[i]-a];
  }
  node[p].god=1;
}
inline void build(){
  q[0]=0;
  for(int i=0,j=0;i<=j;i++){
    for(int l=0;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]:0,
        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]:0;
  }
}
void dfs(int x){
    for(int i=0;i<alpha;i++){
        if(node[node[x].ch[i]].god){
            a[0][x<<1]+=need;
            a[1][(x<<1)+1]+=need;
            a[0][(x<<1)+1]+=need;
            continue;
        }
        a[node[x].ch[i]<<1][x<<1]+=need;
        a[(node[x].ch[i]<<1)+1][(x<<1)+1]+=need;
        if(node[node[x].ch[i]].deep>node[x].deep)
            dfs(node[x].ch[i]);
    }
}
inline void Multi_a(){
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      temp_a[i][j]=0;
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      for(int k=0;k<=size;k++)
        temp_a[i][j]+=a[i][k]*a[k][j];
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      a[i][j]=temp_a[i][j];
}
inline void Multi_b(){
  for(int i=0;i<=size;i++)temp_b[i]=0;
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      temp_b[i]+=a[i][j]*b[j];
  for(int i=0;i<=size;i++)b[i]=temp_b[i];
}
int main(){
  scanf("%d%d%d",&n,&m,&alpha),need=1./alpha;
  for(int i=1;i<=n;i++)
    scanf("%s",s),insert(s);
  build(),dfs(0),size=(sz<<1)+1,b[1]=1.;
  while(m){
    if(m&1)Multi_b();
    m>>=1,Multi_a();
  }
  LD ans=0.;
  for(int i=0;i<=sz;i++)ans+=b[i<<1];
  printf("%lf",(double)ans);
  return 0;
}
只是错了最后一个点

然后我去接受正解,按照我们的思路我们会建出来一个概率矩阵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=100;
char s[20];
int alpha,n,m;
struct Trie{
  int ch[26],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=0;
  for(int i=0;w[i];i++){
    if(node[p].ch[w[i]-a]==0)node[p].ch[w[i]-a]=++sz,node[sz].deep=node[p].deep+1;
    p=node[p].ch[w[i]-a];
  }
  node[p].god=1;
}
inline void build(){
  q[0]=0;
  for(int i=0,j=0;i<=j;i++){
    for(int l=0;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]:0,
        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]:0;
  }
}
void dfs(int x){
    for(int i=0;i<alpha;i++){
        if(node[node[x].ch[i]].god){
            a[0][x]+=need;
            a[sz+1][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=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      temp[i][j]=0;
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      for(int k=0;k<=size;k++)
        temp[i][j]+=a[i][k]*a[k][j];
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      a[i][j]=temp[i][j];
}
inline void Multi_b(){
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      temp[i][j]=0;
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      for(int k=0;k<=size;k++)
        temp[i][j]+=b[i][k]*a[k][j];
  for(int i=0;i<=size;i++)
    for(int j=0;j<=size;j++)
      b[i][j]=temp[i][j];
}
int main(){
  scanf("%d%d%d",&n,&m,&alpha),need=(LD)1./(LD)alpha;
  for(int i=1;i<=n;i++)
    scanf("%s",s),insert(s);
  build(),dfs(0),size=sz+1,a[size][size]=1.;
  for(int i=0;i<=size;i++)b[i][i]=1.;
  while(m){
    if(m&1)Multi_b();
    m>>=1,Multi_a();
  }
  printf("%.6lf",(double)b[size][0]);
  return 0;
}
AC code

 

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

标签:main   opened   技术   none   答案   int   也有   pen   har   

原文地址:http://www.cnblogs.com/TSHugh/p/7528564.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!