BZOJ4820 Sdoi2017 硬币游戏
Description
周末同学们非常无聊，有人提议，咱们扔硬币玩吧，谁扔的硬币正面次数多谁胜利。大家纷纷觉得这个游戏非常符合同学们的特色，但只是扔硬币实在是太单调了。同学们觉得要加强趣味性，所以要找一个同学扔很多很多次硬币，其他同学记录下正反面情况。用H表示正面朝上，用T表示反面朝上，扔很多次硬币后，会得到一个硬币序列。比如HTT表示第一次正面朝上，后两次反面朝上。但扔到什么时候停止呢？大家提议，选出n个同学，每个同学猜一个长度为m的序列，当某一个同学猜的序列在硬币序列中出现时，就不再扔硬币了，并且这个同学胜利，为了保证只有一个同学胜利，同学们猜的n个序列两两不同。很快，n个同学猜好序列，然后进入了紧张而又刺激的扔硬币环节。你想知道，如果硬币正反面朝上的概率相同，每个同学胜利的概率是多少。
Input
第一行两个整数n,m。
接下里n行，每行一个长度为m的字符串，表示第i个同学猜的序列。
1<=n,m<=300
Output
输出n行，第i行表示第i个同学胜利的概率。
输出与标准输出的绝对误差不超过10^-6即视为正确。
Sample Input
3 3
THT
TTH
HTT
Sample Output
0.3333333333
0.2500000000
0.4166666667
#include<bits/stdc++.h>
using namespace std;
#define N 310
#define eps 1e-3
#define For(a,b,c) for(int a=b;a<=c;a++)
int n,m;
double g[N][N];
void guass(){
For(i,1,n){
int p=i;
For(j,i,n)if(abs(g[j][i])>abs(g[p][i]))p=j;
For(j,1,n+1)swap(g[p][j],g[i][j]);
double d=g[i][i];
For(j,1,n+1)g[i][j]/=d;
For(j,1,n){
if(i==j)continue;
double w=g[j][i];
For(k,1,n+1)g[j][k]-=g[i][k]*w;
}
}
For(i,1,n-1)printf("%.10lf
",g[i][n+1]);
}
int fail[N][N];
int s[N][N];
double p[N];
void init(int id){
For(i,2,m){
int j=fail[id][i-1];
while(j&&s[id][i]!=s[id][j+1])j=fail[id][j];
if(s[id][i]==s[id][j+1])fail[id][i]=j+1;
else fail[id][i]=0;
}
}
double kmp(int ida,int idb){
int j=0;
For(i,1,m){
while(j&&s[idb][i]!=s[ida][j+1])j=fail[ida][j];
if(s[idb][i]==s[ida][j+1])++j;
}
double res=0;
while(j){
res+=p[m-j];
j=fail[ida][j];
}
return res;
}
int main(){
scanf("%d%d",&n,&m);
p[0]=1;For(i,1,m)p[i]=p[i-1]*0.5;
For(i,1,n){
static char c[N];
scanf("%s",c+1);
For(j,1,m)
if(c[j]=='H')s[i][j]=1;
else s[i][j]=0;
init(i);
}
For(i,1,n)For(j,1,n)g[i][j]=kmp(i,j);
For(i,1,n)g[i][n+1]=-p[m];
For(i,1,n)g[n+1][i]=1;
g[n+1][n+2]=1;
n++;
guass();
return 0;
}