BZOJ 2553 AC自动机+矩阵快速幂 (神题)

思路:
我们先对所有读进来的T建一个AC自动机
因为走到一个禁忌串就需要回到根
所以呢 搞出来所有的结束点 或一下 fail指针指向的那个点

然后我们就想转移
a[i][j]表示从i节点转移到j节点的概率 如果能够转移到 ans+=1÷alphabet
这里有一个trick
建一个size+1节点 如果回到了根 就连到size+1 a[size+1][size+1]=1
这样就成了累加和了

因为长度最大有10^9,显然直接DP会无论空间还是时间都会爆炸。。。
所以用矩阵乘法+快速幂加速转移
现在考虑怎么处理出初始的转移矩阵
先算出a[i][j]表示i一步到j的概率
用bfs就可以实现,如果j是i的儿子,那么a[i][j]+=1/字符集大小
为了方便我们新建一个节点n=cnt(总结点数)+1
每次转移root时也转移到它
那么a[i][n]就是i走一步匹配到禁忌串的概率。
要把所有步都累加出来,把a[n][n]赋为1就可以了
因为这样下一次计算时b[root][n]=….+b[root][n]*a[n][n]+….
就可以把上次的答案都累加起来了。
自乘x次后,因为贡献永远是1,所以a[root][n]就表示root走x步遇到禁忌串的期望,也就是答案。
http://blog.csdn.net/thy_asdf/article/details/47087113

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 105
#define M 26
using namespace std;
int n,num,len,alphabet,size;
char a[N];
struct matrix{long double a[N][N];void clear(){memset(a,0,sizeof(a));}}st,ans;
matrix operator * (matrix a,matrix b){
    matrix c;c.clear();
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            for(int k=0;k<=n;k++)
                c.a[i][j]+=a.a[i][k]*b.a[k][j];
    return c;
}
struct AC_Automata{
    int ch[N][M],end[N],q[N*M],head,tail,f[N],vis[N];
    void insert(char *s,int num){
        int u=0;
        for(int i=0;s[i];i++){
            int v=s[i]-'a';
            if(!ch[u][v])ch[u][v]=++size;
            u=ch[u][v];
        }end[u]=1;
    }
    void build(){
        f[0]=100;
        while(head<=tail){
            int r=q[head++];
            for(int i=0;i<alphabet;i++){
                int u=ch[r][i];
                if(!u)ch[r][i]=ch[f[r]][i];
                else q[++tail]=u,f[u]=ch[f[r]][i];
            }
            end[r]|=end[f[r]];
        }
        head=tail=q[0]=0,vis[0]=1;
        long double base=1.0/alphabet;
        while(head<=tail){
            int r=q[head++];
            for(int i=0;i<alphabet;i++){
                if(!vis[ch[r][i]])vis[ch[r][i]]=1,q[++tail]=ch[r][i];
                if(end[ch[r][i]])st.a[r][n]+=base,st.a[r][0]+=base;
                else st.a[r][ch[r][i]]+=base;
            }
        }
    }
}ac;
void pow(){for(;len;len>>=1,st=st*st)if(len&1)ans=ans*st;}
int main(){
    scanf("%d%d%d",&num,&len,&alphabet);
    for(int i=1;i<=num;i++)scanf("%s",a),ac.insert(a,i);
    n=size+1;ac.build(),st.a[n][n]=1;
    for(int i=0;i<=n;i++)ans.a[i][i]=1;
    pow();
    printf("%Lf\n",ans.a[0][n]);
}

这里写图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值