BZOJ2553 || 洛谷P4569 [BJWC2011]禁忌【AC自动机+矩阵DP+期望】

Time Limit: 20 Sec
Memory Limit: 128 MBSec

Description

Magic Land上的人们总是提起那个传说：他们的祖先John在那个东方岛屿帮助Koishi与其姐姐Satori最终战平。而后，Koishi恢复了读心的能力
如今，在John已经成为传说的时代，再次造访那座岛屿的人们却发现Koishi遇到了新麻烦。
这次她遇到了Flandre Scarlet——她拥有可以使用禁忌魔法而不会受到伤害的能力。
为了说明什么是禁忌魔法及其伤害，引入以下概念：

1．字母集A上的每个非空字符串对应了一个魔法。其中A是包含了前alphabet个小写字母的集合。
2．有一个集合T，包含了N个字母集A上的字符串T中的每一串称为一个禁忌串（Taboo string）

3．一个魔法，或等价地，其对应的串s因为包含禁忌而对使用者造成的伤害按以下方式确定：
把s分割成若干段，考虑其中是禁忌串的段的数目，不同的分割可能会有不同的数目，其最大值就是这个伤害。

由于拥有了读心的能力，Koishi总是随机地使用Flandre Scarlet的魔法，可以确定的是，她的魔法正好对应字母集A上所有长度为len的串。

但是，Flandre Scarlet所使用的一些魔法是带有禁忌的，由于其自身特性，她可以使用禁忌魔法而不受到伤害，而Koishi就不同了。可怜的Koishi每一次使用对方的魔法都面临着受到禁忌伤害的威胁。

你现在需要计算的是如果Koishi使用对方的每一个魔法的概率是均等的，那么每一次随机使用魔法所受到的禁忌伤害的期望值是多少。

Input

第一行包含三个正整数N、len、alphabet。
接下来N行，每行包含一个串Ti，表示禁忌串。

Output

一个非负实数，表示所受到禁忌伤害的期望值。

HINT

100%的数据中N ≤ 5，len ≤109，1 ≤ alphabet ≤ 26。
在所有数据中，有不少于40%的数据中：N = 1。
数据保证每个串Ti的长度不超过15，并且不是空串。
数据保证每个Ti均仅含有前alphabet个小写字母。
数据保证集合T中没有相同的元素，即对任意不同的i和j，有Ti≠Tj。
【评分方法】
对于每一组数据，如果没有得到正确的输出（TLE、MLE、RTE、输出格式错误等）得0分。
否则：设你的输出是YourAns，标准输出是StdAns：
记MaxEPS = max(1.0 , StdAns)×10-6
如果|YourAns – StdAns| ≤ MaxEPS则得10分，否则得0分。
即：你的答案需要保证相对误差或绝对误差不超过10-6

---

题目分析

多串匹配，AC自动机
每个单词的贡献值是1，所以计算期望其实转化为计算概率
$dp[i][j]$表示从AC自动机结点i走到结点j的概况
转移的时候遇到单词结点，因为要求禁忌串是分割开的，所以此时回到根节点

由于len很大，所以考虑矩阵快速幂优化
注意多建一个虚拟点，若遇到u转移的儿子是单词结点，那么矩阵中u到根，u到虚点都加上相应概率
最后方便矩乘后得到答案

毒瘤卡精度，要开long double，还第一次见识到了cout控制浮点数位数的方法

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<algorithm>
#include<cstring>
#include<iomanip>
using namespace std;
typedef long long lt;
typedef long double dd;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int maxn=155;
int n,L,alp;
int ch[maxn][26],rem[maxn];
int fail[maxn],sz;
char pt[maxn];

struct matrix{
	int row,col;
    dd mat[maxn][maxn]; 
    matrix(int r=0,int c=0){ 
    	row=r; col=c;
		memset(mat,0,sizeof(mat));
	}
}h;

matrix operator *(matrix a,matrix b){ 
	matrix c=matrix(a.row,b.col);
	for(int i=0;i<=a.row;++i)
	for(int j=0;j<=b.col;++j)
	for(int k=0;k<=a.col;++k)
	c.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
	return c;
}

matrix qpow(matrix a,lt k)
{
	matrix res=matrix(a.row,a.col);
	for(int i=0;i<=a.row;++i) res.mat[i][i]=1;
	while(k){
		if(k&1) res=res*a;
		a=a*a; k>>=1;
	}
	return res;
}

void ins(char* ss,int len)
{
    int u=0;
    for(int i=0;i<len;++i)
    {
        int x=ss[i]-'a';
        if(!ch[u][x]) ch[u][x]=++sz;
        u=ch[u][x];
    }
    rem[u]=1;
}

void ACM()
{
    queue<int> q;
    for(int i=0;i<alp;++i)
    if(ch[0][i]) fail[ch[0][i]]=0,q.push(ch[0][i]);
    while(!q.empty())
    {
        int u=q.front(); q.pop();
        for(int i=0;i<alp;++i)
        {
            if(!ch[u][i]) ch[u][i]=ch[fail[u]][i];
            else{
                fail[ch[u][i]]=ch[fail[u]][i];
                q.push(ch[u][i]);
                rem[ch[u][i]]|=rem[fail[ch[u][i]]];
            }
        }
    }
}

void build()
{
    dd p=(dd)1.0/(dd)alp;
    h.row=h.col=sz+1;
    for(int i=0;i<=sz;++i)
    {
        if(rem[i]) continue;
        for(int j=0;j<alp;++j)
        {
            if(rem[ch[i][j]]){
                h.mat[i][sz+1]+=p;
                h.mat[i][0]+=p;
            }
            else h.mat[i][ch[i][j]]+=p;
        }
    }
    h.mat[sz+1][sz+1]=(dd)1.0;
}

int main()
{
    n=read();L=read();alp=read();
    for(int i=1;i<=n;++i)
    {
        scanf("%s",&pt);
        ins(pt,strlen(pt));
    }
    
    ACM(); build();
    matrix res=qpow(h,L);
    
    cout<<fixed<<setprecision(8)<<res.mat[0][sz+1]<<endl;
    return 0;
}