zoj 3754 One Person Game（期望dp）

-Dong

于 2019-07-15 16:34:50 发布

阅读量178

点赞数

分类专栏：期望/概率dp

本文链接：https://blog.csdn.net/LJD201724114126/article/details/95979077

版权

期望/概率dp 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

题目链接：哆啦A梦传送门

题意：有三个骰子，分别有k1，k2，k3面，初始分数为0，每摇动骰子后，如果三个面分别为a，b，c，那么将分数置零，否则加上a+b+c的分数，当分数大于n时结束。问：期望次数是多少次？

题解：

参考博客：

设dp[i]表示分数为i，到达目标状态需要的期望，p[j]表示摇骰子分时为j的概率，p0表示初始分数。

那么我们可以得到转移方程：

$\large dp[i]=\sum_{j=1\Lambda j!=a+b+c}^{k1+k2+k3}(dp[i+j]*p[j])+dp[0]*p_0+1$

很显然我们要求的是dp[0]，但是递推式右边就已经包含了dp[0]，我们就不能直接算出来。

我们知道dp[0]为常数，此转移方程为线性递推的，所以我们可以这样设：

$\large dp[i]=A[i]*dp[0]+B[i]$

即： $\large dp[i+j]=A[i+j]*dp[0]+B[i+j]$

将这个式子带入第一条式子，可得：

$dp[i]=(\sum_{j=1\Lambda j!=a+b+c}^{k1+k2+k3}A[i+j]*p[j]+p_0)dp[0]+\sum_{j=1\Lambda j!=a+b+c}^{k1+k2+k3}B[i+j]*p[j]+1$

得：

$\large A[i]=\sum_{j=1\Lambda j!=a+b+c}^{k1+k2+k3}A[i+j]*p[j]+p_0$

$B[i]=\sum_{j=1\Lambda j!=a+b+c}^{k1+k2+k3}B[i+j]*p[j]+1$

故最后 $dp[0]=\frac{B[0]}{1-A[0]}$

#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

double A[610],B[610];
double p[100];

int main()
{
        int n,k1,k2,k3,a,b,c;
        int ncase;
        scanf("%d",&ncase);

        while(ncase--)
        {
            scanf("%d%d%d%d%d%d%d",&n,&k1,&k2,&k3,&a,&b,&c);

            memset(p,0,sizeof(p));
            memset(A,0,sizeof(A));
            memset(B,0,sizeof(B));

            double p0=1.0/(double)(k1*k2*k3);
            for(int i=1;i<=k1;i++)
                for(int j=1;j<=k2;j++)
            for(int k=1;k<=k3;k++){
                if(i==a&&j==b&&k==c){
                    continue;
                }
                else{
                    p[i+j+k]+=p0;
                }
            }


            for(int i=n;i>=0;i--)
            {
                A[i]=p0;B[i]=1;
                for(int j=1;j<=k1+k2+k3;j++)
                {
                    A[i]+=A[i+j]*p[j];
                    B[i]+=B[i+j]*p[j];
                }
            }
            printf("%.16f\n",B[0]/(1.0-A[0]));
        }

        return 0;
}