您的位置:首页 > 其它

zoj 3329 One Person Game(概率dp)

2014-03-18 11:33 337 查看
题意:三个骰子,这三个骰子有K1、K2、K3个面,cnt最开始等于0,同时掷3个骰子,如果这三个骰子分别为a,b,c,那么cnt=0,否则cnt+=(a+b+c)。如果cnt<=n,那么接着掷,求掷骰子的次数的期望。

思路:最开始推完方程以后发现存在一个E(0)的常数,本来想用直接用dp[i]表示dp[i]-E[0],后来发现不行,因为最后边界的位置就错了。正确的姿势是把E(i)表示为E(i)=A[i]*E[0]+B[i],然后转化为求A[i]和B[i]。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=500+100;
double A[maxn],B[maxn],p;
bool flag[maxn];
int a,b,c,n,k1,k2,k3;
void f(int x)
{
    if(x>n) {A[x]=B[x]=0;return ;}
    if(flag[x]) return ;
    flag[x]=true;
    A[x]=p;B[x]=1;
    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;
                f(x+i+j+k);
                A[x]+=p*A[x+i+j+k];
                B[x]+=p*B[x+i+j+k];
            }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d%d%d%d%d",&n,&k1,&k2,&k3,&a,&b,&c);
        memset(flag,0,sizeof(flag));
        p=1.0/k1/k2/k3;
        f(0);
        printf("%.16lf\n",B[0]/(1-A[0]));
    }
    return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: