Zoj 3329 (dp求期望)

第二个dp求数学期望的题，如果看不懂，请看我的第一个dp求期望的题（Poj 2096）

/**
    dp求期望的题。
    题意：
    有三个均匀的骰子，分别有k1,k2,k3个面，初始分数是0，
    当掷三个骰子的点数分别为a,b,c的时候，分数清零，否则分数加上三个骰子的点数和，
    当分数>n的时候结束。求需要掷骰子的次数的期望。
    题解:
    设 E[i]表示现在分数为i，到结束游戏所要掷骰子的次数的期望值。
    显然 E[>n] = 0; E[0]即为所求答案;
    E[i] = ∑Pk*E[i+k] + P0*E[0] + 1; (Pk表示点数和为k的概率，P0表示分数清零的概率)
    由上式发现每个 E[i]都包含 E[0]，而 E[0]又是我们要求的，是个定值。
    设 E[i] = a[i]*E[0] + b[i];
    将其带入上面的式子：
    E[i] = ( ∑Pk*a[i+k] + P0 )*E[0] + ∑Pk*b[i+k] + 1;
    显然，
    a[i] = ∑Pk*a[i+k] + P0;
    b[i] = ∑Pk*b[i+k] + 1;
    当 i > n 时：
    E[i] = a[i]*E[0] + b[i] = 0;
    所以 a[i>n] = b[i>n] = 0;
    可依次算出 a[n],b[n]; a[n-1],b[n-1] ... a[0],b[0];
    则 E[0] = b[0]/(1 - a[0]);
**/

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

int main()
{
    int    nc, n, ks, k1, k2, k3, a, b, c;
    double p0, p[20];

    cin >> nc;
    while ( nc-- )
    {
        cin >> n >> k1 >> k2 >> k3 >> a >> b >> c;
        ks = k1 + k2 + k3;
        p0 = 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 )
                        p[i+j+k] += p0;
                }

        double a[520] = {0}, b[520] = {0};
        for (int i = n; i >= 0; i--)
        {
            for (int k = 3; k <= ks; k++)
            {
                a[i] += a[i+k]*p[k];
                b[i] += b[i+k]*p[k];
            }
            a[i] += p0;
            b[i] += 1;
        }
        printf("%.15lf\n", b[0]/(1 - a[0]) );
    }
    return 0;
}