P3706 [SDOI2017] 硬币游戏

最新推荐文章于 2024-10-11 12:01:25 发布

23010359

最新推荐文章于 2024-10-11 12:01:25 发布

阅读量332

点赞数 6

文章标签：算法 c++

本文链接：https://blog.csdn.net/yfuffdigtybcg/article/details/135478135

版权

#include<bits/stdc++.h>
using namespace std;
int s[305][305];
char ss[305];
int n,m;
double A[303][303],p[303],res[303];
int has[303][303],bb[303];
const int base=13131;
const int mod=1e9+7;
void guass(int n)
{
    for(int i=1;i<=n;i++)
    {
        int pos=i;
        for(int j=i;j<=n;j++) if(fabs(A[j][i])>fabs(A[pos][i])) pos=i;
        for(int j=1;j<=n+1;j++) swap(A[pos][j],A[i][j]);
        for(int j=1;j<=n;j++)
        {
            if(i!=j)
            {
                double t=A[j][i]/A[i][i];
                for(int k=i+1;k<=n+1;k++) A[j][k]-=A[i][k]*t;
            }
        }
    }
    for(int i=1;i<=n;i++) res[i]=A[i][n+1]/A[i][i];
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",ss+1);
        for(int j=1;j<=m;j++)
            s[i][j]=(ss[j]=='T'?0:1);
    }
    p[0]=1.0; bb[0]=1;
    for(int i=1;i<=m;i++) p[i]=p[i-1]*0.5;
    for(int i=1;i<=m;i++) bb[i]=bb[i-1]*base;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            has[i][j]=1LL*has[i][j-1]*base+s[i][j];
    for(int i=1;i<=n;i++)
    {
        A[i][n+1]=-p[m]; A[i][i]=1.0;
        for(int j=1;j<=n;j++)
        {
            for(int len=1;len<=m-(i==j);len++)
                if(has[i][len]==has[j][m]-has[j][m-len]*bb[len])
                    A[i][j]+=p[m-len];
        }
    }
    for(int i=1;i<=n;i++)
        A[n+1][i]=1.0;
    A[n+1][n+2]=1.0;
//    for(int i=1;i<=n+1;i++)
//    {
//        for(int j=1;j<=n+1;j++)
//            printf("%.3f ",A[i][j]);
//        printf("= %.3f\n",A[i][n+2]);
//    }
    guass(n+1);
    for(int i=1;i<=n;i++) printf("%.10f\n",res[i]);
    return 0;
}