zoj-3329-One Person Game-有环的概率DP

感觉kuangbin写的题解不错，就直接抄袭过来啦。。。

地址：http://www.cnblogs.com/kuangbin/archive/2012/10/03/2710648.html

题意：有三个骰子，分别有k1,k2,k3个面。
每次掷骰子，如果三个面分别为a,b,c则分数置0，否则加上三个骰子的分数之和。
当分数大于n时结束。求游戏的期望步数。初始分数为0


设dp[i]表示达到i分时到达目标状态的期望，pk为投掷k分的概率，p0为回到0的概率
则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
     先递推求得A[0]和B[0].
     那么  dp[0]=B[0]/(1-A[0]);

#include <iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
using namespace std;
#define maxn 1001
#define eps 1e-6
#define zero(x) (fabs(x)<0?0:x)
double p[7*7*7];
double a[maxn];
double b[maxn];
int main()
{
    double x,y,p0;
    int n,m,T;
    scanf("%d",&T);
    int k1,k2,k3;
    int aa,bb,cc;
    while(T--)
    {
        scanf("%d",&n);
        scanf("%d%d%d",&k1,&k2,&k3);
        scanf("%d%d%d",&aa,&bb,&cc);
        int m=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++)
                {
                     p[i+j+k]+=1.0/m;
                }
            }
        }
        p0=1.0/m;
        p[aa+bb+cc]-=p0;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i=n;i>=0;i--)
        {
            for(int j=1;j<=k1+k2+k3;j++)
            {
                a[i]+=p[j]*a[i+j];
                b[i]+=p[j]*b[i+j];
            }
            a[i]+=p0;
            b[i]+=1.0;
        }
        printf("%.10f\n",b[0]/(1-a[0]));
    }
    return 0;
}