【bzoj2553】禁忌 AC自动机+矩阵乘法

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2553 

【题解】

这是一道神题，出题人卡精度什么的就不说了。

首先把模式串建成AC自动机（或trie图），然后考虑在AC自动机上的转移。

对于每一步转移，都有两种情况：

1、子结点没被标记（即不是模式串的结尾单词），有1/alphabet的期望转移到这个子结点。

2、子结点被标记，有1/alphabet的期望值转移到根，并且用一个新结点记录答案。

这样我们可以考虑构造一个矩阵a[i][j]，记录下来结点之间的转移关系。

然后矩阵自乘m次得到的矩阵ans，ans[0][cnt+1]就是答案。

解释一下：对于从i结点转移到j结点，存在一个结点k，使得从i转移到k，然后再转移到j，这就是矩阵乘法的意义。

看了zyf的代码，学到了一种写矩阵乘法的好方法：把矩阵存成结构体，然后定义一种矩阵乘运算，很效率。

调试的时候遇到了一个问题，这题必须用long double,然而long double数组开到了417*417就会出现很奇怪的现象（读者可以自行尝试），只能开到416*416.

具体见代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAXN 205
struct node{long double p[MAXN][MAXN];node(){memset(p,0,sizeof(p));}}a,ans;
int n,m,K,cnt,end[MAXN],q[MAXN],fail[MAXN],vis[MAXN],tr[MAXN*5][27];
char ch[MAXN];
inline int read()
{
    int x=0,f=1;  char ch=getchar();
    while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getchar();}
    while(isdigit(ch))  {x=x*10+ch-'0';  ch=getchar();}
    return x*f;
}
inline node operator *(node &x,node &y)//定义矩阵乘运算
{
    node z;
    for(int i=0;i<=cnt+1;i++)
        for(int j=0;j<=cnt+1;j++)
            for(int k=0;k<=cnt+1;k++)
                z.p[i][j]+=x.p[i][k]*y.p[k][j];
    return z;
}
void insert()
{
    int now=0,len=strlen(ch+1);  
    for(int i=1;i<=len;i++)
    {
        if(!tr[now][ch[i]-'a'])  tr[now][ch[i]-'a']=++cnt;
        now=tr[now][ch[i]-'a'];
    }
    end[now]=1;
}
void build()
{
    int head=0,tail=0;
    for(int i=0;i<K;i++) if(tr[0][i]) q[++tail]=tr[0][i];
    while(++head<=tail)
    {
        int x=q[head];
        for(int i=0;i<K;i++)
        {
            if(!tr[x][i])  tr[x][i]=tr[fail[x]][i];
            else {fail[tr[x][i]]=tr[fail[x]][i];  q[++tail]=tr[x][i];}
        }
    }
}
void get()
{
    int head=0,tail=1; long double chty=(long double)1.0/K;//期望常数
    q[1]=0; vis[0]=1;
    while(++head<=tail)
    {
        int x=q[head];
        for(int i=0;i<K;i++)
        {
            if(!vis[tr[x][i]])  vis[tr[x][i]]=1,q[++tail]=tr[x][i];
            if(end[tr[x][i]])  a.p[x][cnt+1]+=chty,a.p[x][0]+=chty;
            else a.p[x][tr[x][i]]+=chty;
        }
    }
    a.p[cnt+1][cnt+1]=1;
}
int main()
{
    //freopen("cin.in","r",stdin);
    //freopen("cout.out","w",stdout);
    n=read();  m=read();  K=read();
    for(int i=1;i<=n;i++)  {scanf("%s",ch+1);  insert();}
    build(); 
    get();
    for(int i=0;i<=cnt+1;i++)  ans.p[i][i]=1;
    for(;m;m>>=1,a=a*a)  if(m&1) ans=ans*a;
    printf("%.7f\n",(double)ans.p[0][cnt+1]);
    return 0;
}