【BZOJ4820】【SDOI2017】硬币游戏

Description

　　
　　
　　

Solution

　　
　　设当前走出了一个不匹配任何字符串的串(S)。
　　
​　　若在(S)后随机增添(m)个字符，单看这些字符而言，这(m)个字符匹配到每个玩家的字符串的概率是相同的，记为(P)。
　　
　　问题在于，对于每个字符串来说，并不是所有情况下一定要通过新增添(m)个字符才能匹配到自己，有可能加到中途时，就已经与(S)的某个后缀组成了自己，又或者是与(S)的某个后缀组成了别的字符串，早该停止了。
　　
　　但是，对于每个串，不管每种情况中途该不该停下，我们计算出每种情况继续增加满(m)个字符并匹配到自己的概率，它们的概率之和还是(P)。
　　
　　 记每个人(i)成功被匹配到的概率是(f_i)（答案的定义）。
　　
　　现在枚举对于一个人(i)，在新加(m)个字符尝试匹配自己时，所有中途应该停下的情况。
　　
　　​ 枚举另一个人(j)，如果(j)的(len)后缀与(i)的(len)前缀相同，则配合(S)的随机性，可能出现了这种情况：
　　

　　
​　　 这时候早就该停了，但为了凑齐(P)，要计算在这种情况下继续匹配完全时所需的概率。继续匹配完(i)的子串，则还需要((frac 1 2)^{m-len})的概率。因此，这种情况对总和的贡献有((frac 1 2)^{m-len}f_j)。当然，(i)和(j)之间不止有一个(len)满足条件，应找出所有符合描述的(len)，累加(f_j)的贡献系数，记最终(f_j)贡献系数为(a_j=sum(frac 1 2)^{m-len})。
　　
​　　 其实(j)可以等于(i)，这代表着提前匹配到自己的情况。
　　
​　　 我们可以列出等式：

[P=sum_{i=1}^na_if_i ]

​　　 总的来说，每一个人获胜的概率之和应该是1，因此有等式

[sum_{i=1}^nf_i=0 ]

​　　 算上(P)，我们拿到了一个(n+1)个未知数的(n+1)条方程，高斯消元解决即可，尽管我们并不需要知道(P)的具体取值。
　　　　
　　
　　

Code

　　

#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
const int N=305;
int n,m;
char str[N][N];
double a[N][N],x[N];
int nex[N*2];
double mi2[N];
double kmp(int x,int y){
	static char b[N*2];
	for(int i=1;i<=m;i++) b[i]=str[x][i],b[m+i]=str[y][i];
	nex[1]=0;	
	for(int i=2,j;i<=m*2;i++){
		j=nex[i-1];
		while(j&&b[j+1]!=b[i]) j=nex[j];
		if(b[j+1]==b[i]) nex[i]=j+1;
		else nex[i]=0;
	}
	double res=0;
	for(int i=m*2;i;i=nex[i])
		if(i<=m) res+=mi2[m-i];
	return res;
}
void fill_matrix(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++) 
			a[i][j]=kmp(i,j);
		a[i][n+1]=-1;
	}
	for(int i=1;i<=n;i++) a[n+1][i]=1;
	a[n+1][n+2]=1;
}
void gaussian(int n){
	int best;
	for(int i=1;i<=n;i++){
		best=i;
		for(int j=i+1;j<=n;j++)
			if(fabs(a[j][i])>fabs(a[best][i])) best=j;
		if(best!=i)
			for(int j=i;j<=n+1;j++) swap(a[i][j],a[best][j]);
		for(int j=i+1;j<=n;j++){
			double t=a[j][i]/a[i][i];
			for(int k=i;k<=n+1;k++)
				a[j][k]-=a[i][k]*t;
		}
	}
	for(int i=n;i>=1;i--){
		for(int j=i+1;j<=n;j++) a[i][n+1]-=a[i][j]*x[j];
		x[i]=a[i][n+1]/a[i][i];
	}
}
int main(){
	scanf("%d%d",&n,&m);
	mi2[0]=1;
	for(int i=1;i<=m;i++) mi2[i]=mi2[i-1]*0.50000000000;
	for(int i=1;i<=n;i++) scanf("%s",str[i]+1);
	fill_matrix();
	gaussian(n+1);
	for(int i=1;i<=n;i++) printf("%.10lf
",x[i]);
	return 0;
}

​

​