BZOJ2553 Beijing2011禁忌(AC自动机+动态规划+矩阵快速幂+概率期望)

时间:2023-03-09 17:51:23
BZOJ2553 Beijing2011禁忌(AC自动机+动态规划+矩阵快速幂+概率期望)

  考虑对一个串如何分割能取得最大值。那么这是一个经典的线段覆盖问题,显然每次取右端点尽量靠前的串。于是可以把串放在AC自动机上跑,找到一个合法串后就记录并跳到根。

  然后考虑dp。设f[i][j]表示前i位走到AC自动机上j节点的概率,枚举下个字符即可转移。同时记录此时期望伤害,找到合法串就统计入答案。

  并且注意到每次转移是相同的。矩阵快速幂优化即可。

  以及非常卡精度,需要全程long double。cout的保留小数位数误差是相当大的,必须用printf。并且转移到某个字符的概率即1/alphabet需要强制转换成long double,否则也会丢失精度。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
int x=,f=;char c=getchar();
while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
return x*f;
}
#define N 80
#define double long double
int n,m,S,trie[N][],fail[N],val[N],q[N],cnt=;
char c[];
struct matrix
{
int n;double a[N][N];
matrix operator *(const matrix&b) const
{
matrix c;c.n=n;memset(c.a,,sizeof(c.a));
for (int i=;i<n;i++)
for (int j=;j<N;j++)
for (int k=;k<N;k++)
c.a[i][j]+=a[i][k]*b.a[k][j];
return c;
}
}f,a;
void ins(char *a)
{
int n=strlen(a+),t=;
for (int i=;i<=n;i++)
{
if (!trie[t][a[i]-'a']) trie[t][a[i]-'a']=++cnt;
t=trie[t][a[i]-'a'];
}
val[t]=;
}
void build()
{
int head=,tail=;for (int i=;i<S;i++) if (trie[][i]) q[++tail]=trie[][i];
do
{
int x=q[++head];
for (int i=;i<S;i++)
if (trie[x][i]) q[++tail]=trie[x][i],fail[trie[x][i]]=trie[fail[x]][i],val[trie[x][i]]|=val[fail[trie[x][i]]];
else trie[x][i]=trie[fail[x]][i];
}while (head<tail);
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("bzoj2553.in","r",stdin);
freopen("bzoj2553.out","w",stdout);
const char LL[]="%I64d\n";
#else
const char LL[]="%lld\n";
#endif
n=read(),m=read(),S=read();
for (int i=;i<=n;i++) scanf("%s",c+),ins(c);
build();
a.n=cnt+;
for (int i=;i<=cnt;i++)
for (int j=;j<S;j++)
if (val[trie[i][j]]) a.a[i][]+=(double)/S,a.a[i][cnt+]+=(double)/S;
else a.a[i][trie[i][j]]+=(double)/S;
a.a[cnt+][cnt+]=;
f.n=;f.a[][]=;
for (;m;a=a*a,m>>=) if (m&) f=f*a;
printf("%.8Lf",f.a[][cnt+]);
return ;
}