ZOJ 3329 One Person Game （概率DP）

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3329

题目：抛掷3个骰子（分别共有k1,k2,k3个点)，如果对应的点数分别是a,b,c那么就变成0，累加加上点数之和，直到score>n时结束。问投掷3个骰子的次数。

做完这题发现，原来，一般的概率DP的设计都是从后向前推导。

设dp[i]是分数为i的次数期望， p[i]是相关概率，有：
  ( i+k!=0 ) （1）
抽象成：    （2）
（dp[i+k]含有dp[0]，所以不能简单地认为a[0]就是p[0]）
将     带入（1）计算，得到：

联系（2）得到：

这样一带一比就找到了a[i]和b[i]的递推式，很神奇啊有木有
当i=0时，有

因为dp(i>n)=0，所以有 a[i>n]=b[i>n]=0。
最后求得dp[0]即可。

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