bzoj2553【beijing2011】禁忌

题意：http://www.lydsy.com/JudgeOnline/problem.php?id=2553

sol  ：puts("nan"); (逃~

　　　ac自动机+矩阵快速幂

　　　先将所有的串放到ac自动机上，贪心匹配，对一个包含禁忌串的节点划分出一段

　　　设f[i][j]表示走了i步到达AC自动机上的j节点的概率

　　　转移方程为f[i+1][k]=f[i][j]/alphabet

　　　由于i较大，但是每一步的转移是一样的，所以可以用矩阵快速幂优化

　　　P.S.喜sang闻xin乐bing见kuang的出题人卡精度QAQ，需要开long double

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int Mx=;
struct Node { int l,r; long double f[Mx][Mx]; } str,final;
int n,m,cnt,len,num,ch[Mx][],fail[Mx],q[Mx],jud[Mx];
char s[Mx];
void build_trie()
{
    int x=;
    for(int i=;i<=m;i++)
    {
        if(!ch[x][s[i]-'a']) ch[x][s[i]-'a']=++cnt;
        x=ch[x][s[i]-'a'];
        if(jud[x]) break;
    }
    jud[x]=;
}
void get_fail()
{
    int h=,t=;q[h]=;
    while(h!=t)
    {
        int now=q[h++]; if(h>Mx) h=;
        for(int i=;i<num;i++)
            if(ch[now][i])
            {
                int x=ch[now][i],y=fail[now];
                while(y&&(!ch[y][i])) y=fail[y];
                if(ch[y][i]!=x) fail[x]=ch[y][i];
                else fail[x]=;
                q[t++]=x; if(t>Mx) t=;
            }
            else ch[now][i]=ch[fail[now]][i];
    }
    str.f[cnt+][cnt+]=;
}
Node mul(Node x,Node y)
{
    Node tmp;tmp.l=x.l;tmp.r=y.r;
    memset(tmp.f,,sizeof(tmp.f));
    for(int k=;k<x.r;k++)
        for(int i=;i<x.l;i++)
            for(int j=;j<y.r;j++)
                tmp.f[i][j]+=x.f[i][k]*y.f[k][j];
    return tmp;
}
void build()
{
    get_fail();
    long double tmp=(long double) /num;
    for(int i=;i<=cnt;i++)
        if(!jud[i])
            for(int j=;j<num;j++)
                if(jud[ch[i][j]]) str.f[i][cnt+]+=tmp,str.f[i][]+=tmp;
                else str.f[i][ch[i][j]]+=tmp;
    str.f[cnt+][cnt+]=;
}
void dfs(int x)
{
    if(jud[x])
        for(int i=;i<num;i++) ch[x][i]=;
    else
        for(int i=;i<num;i++)
            if(ch[x][i]) dfs(ch[x][i]);
}
void pre()
{
    scanf("%d%d%d",&n,&len,&num);
    for (int i=;i<=n;i++)
    {
        scanf("%s",s+); m=strlen(s+);
        build_trie();
    }
    dfs();
    str.l=str.r=cnt+;
    final.l=,final.r=cnt+,final.f[][]=;
}
int main()
{
    pre();
    build();
    for( ;len;len>>=,str=mul(str,str)) if(len&) final=mul(final,str);
    printf("%.9lf",(double) final.f[][cnt+]);
}