BZOJ4820 [SDOI2017]硬币游戏

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 305

using 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;
}