BZOJ4820 [SDOI2017]硬币游戏

题面：BZOJ

解析

考虑把所有没有到达结束状态的字符串看做一类字符串\(N\)，把以字符串\(i\)作为结束的一类字符串\(i\)。现在假设猜测了两个字符串\(A=TTH\)，\(B=HTT\)，不难发现可以列一个方程出来：

\[P(NTTH)=P(A)\]

遗憾的是，这个方程是错的，因为在\(N\)向后匹配的过程中，有可能先一步匹配其他字符串，比如\(N\)以\(HT\)结尾，只要加上一个\(T\)，就以\(B\)结束了，我们试着把它写进方程里面，有：

\[P(NTTH)=P(A)+P(BTH)+P(BH)\]

写作带系数的形式：

\[0.125P(N)=P(A)+0.25P(B)+0.5P(B)\]

不难发现只要一个字符串\(B\)的后缀为字符串\(A\)的前缀，在\(A\)对应的方程里就有一个贡献，这个不难，随便用个字符串算法就能处理。

那么我们可以对\(n\)个字符串分别列出方程，再加上所有\(P(i)\)概率之和为1，一共\(n+1\)个方程，\(n+1\)个变量，高斯消元即可。

代码

#include<cmath>
#include<queue>
#include<cstdio>
#define N 305using namespace std;int n,m,s[N][N]; char str[N];inline int In(){char c=getchar(); int x=0,ft=1;for(;c<'0'||c>'9';c=getchar()) if(c=='-') ft=-1;for(;c>='0'&&c<='9';c=getchar()) x=x*10+c-'0';return x*ft;
}int h[N*N],e_tot=0;
struct E{ int to,nex; }e[N*N];inline void add(int u,int v){e[++e_tot]=(E){v,h[u]}; h[u]=e_tot;
}int rt=0,T_tot=0,ch[N*N][2],fail[N*N],d[N*N];inline void Insert(int i){int u=rt;for(int j=0;j<m;++j){if(!ch[u][s[i][j]]) ch[u][s[i][j]]=++T_tot;u=ch[u][s[i][j]]; add(u,i);}
}inline void Get_fail(int m){queue<int> Q; fail[rt]=rt;for(int i=0,v;i<=m;++i){v=ch[rt][i]; if(!v) continue;Q.push(v); fail[v]=rt; d[v]=1;}while(!Q.empty()){int u=Q.front(); Q.pop();for(int i=0,p,v;i<=m;++i){v=ch[u][i]; if(!v) continue;Q.push(v); p=fail[u];while(p&&!ch[p][i]) p=fail[p];fail[v]=ch[p][i]; d[v]=d[u]+1;}}
}double a[N][N],p[N],ans[N];inline void Get_v(int u,int B){if(!u) return;for(int i=h[u];i;i=e[i].nex) a[e[i].to][B]-=p[m-d[u]];Get_v(fail[u],B);
}inline void Get_a(){for(int i=1;i<=n;++i) a[i][n+1]=p[m];for(int i=1,u;i<=n;++i){u=rt;for(int j=0;j<m;++j){while(u&&!ch[u][s[i][j]]) u=fail[u];u=ch[u][s[i][j]];}Get_v(u,i);}for(int i=1;i<=n;++i) a[n+1][i]=1.0; a[n+1][n+2]=1.0;
}inline void Gauss(int n){for(int i=1,q;i<=n;++i){q=i;for(int j=i+1;j<=n;++j)if(fabs(a[j][i])>fabs(a[q][i])) q=j;swap(a[i],a[q]);for(int j=n+1;j>=i;--j)a[i][j]=a[i][j]/a[i][i];for(int j=i+1;j<=n;++j)for(int k=n+1;k>=i;--k)a[j][k]-=a[i][k]*a[j][i];}for(int i=n;i;--i){ans[i]=a[i][n+1];for(int j=i+1;j<=n;++j)ans[i]-=a[i][j]*ans[j];}
}int main(){n=In(); m=In(); p[0]=1.0;for(int i=1;i<=m;++i) p[i]=0.5*p[i-1];for(int i=1;i<=n;++i){scanf("%s",str);for(int j=0;j<m;++j) s[i][j]=(str[j]=='H');Insert(i);}Get_fail(1); Get_a(); Gauss(n+1);for(int i=1;i<=n;++i) printf("%.10lf\n",ans[i]);return 0;
}