ZOJ 3329 One Person Game期望dp

最新推荐文章于 2023-07-04 08:00:00 发布

qiao111_

最新推荐文章于 2023-07-04 08:00:00 发布

阅读量106

点赞数 1

分类专栏：概率dp

本文链接：https://blog.csdn.net/qiao111_/article/details/97672843

版权

概率dp 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

https://vjudge.net/contest/312931#problem/D
题意
三个骰子，分别有k1,k2,k3个面。
每次掷骰子，如果三个面分别为a,b,c则分数置0，否则加上三个骰子的分数之和。
当分数大于n时结束。求游戏的期望步数。初始分数为0
题解
设dp[i]表示达到i分时到达目标状态的期望，pk为投掷k分的概率，p0为回到0的概率
dp[n] = p0 * dp[0] + 1
dp[n-1] = p0 * dp[0] + p1 * dp[n] + 1

则dp[i]=∑(pk * dp[i+k])+dp[0] * p0+1;
都和dp[0]有关系，而且dp[0]就是我们所求，为常数
设dp[i]=A[i] * dp[0]+B[i];
代入上述方程右边得到：
dp[i]=∑(pk * A[i+k] * dp[0]+pk * B[i+k])+dp[0] * p0+1
=(∑(pk * A[i+k])+p0)dp[0]+∑(pk * B[i+k])+1;
则A[i]=(∑(pk * A[i+k])+p0)
B[i]=∑(pk * B[i+k])+1
则dp[0]=B[0]/(1-A[0])
代码

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
double p[50];
double A[505],B[505];
int main()
{
    int t;
    int n,k1,k2,k3,a,b,c;
    scanf("%d",&t);
    while(t--)
    {
        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));
        p[0]=1.0/(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;
                    p[i+j+k]+=p[0];
                }
        A[n]=p[0];
        B[n]=1;
        for(int i=n-1; i>=0; i--)
        {
            A[i]=p[0];
            B[i]=1.0;
            for(int j=1;j<=k1+k2+k3&&i+j<=n; j++)
            {
                A[i]+=A[i+j]*p[j];
                B[i]+=B[i+j]*p[j];
            }
        }
        printf("%.15f\n",B[0]/(1.0-A[0]));
    }
    return 0;
}