ZOJ 3329 One Person Game

分析转自here

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

分析：本题通过代换系数，化简后求系数。
一般形成环的用高斯消元法求解。但是此题都是和dp[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+k]=A[i+k]*dp[0]+B[i+k]代入上述方程右边得到：
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

当i>n时：dp[i]=A[i]*dp[0]+B[i]=0,所以A[i>n]=B[i>n]=0。

所以可以先递推求得A[0]和B[0].那么  dp[0]=B[0]/(1-A[0]);

Code :

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
#include <map>
#include <set>
#define eps 1e-6
#define LL long long
#define pb push_back
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;

const int maxn=555;
double dp[maxn],A[maxn],B[maxn],p[55];

int main()
{
    int T,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);
        int ks=k1+k2+k3;
        double p0=1.0/(k1*k2*k3);
        memset(dp,0,sizeof(dp));
        memset(A,0,sizeof(A));
        memset(B,0,sizeof(B));
        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]+=p0;
                }
            }
        }
        for(int i=n;i>=0;i--){
            for(int k=3;k<=ks;k++){
                A[i]+=p[k]*A[i+k];
                B[i]+=p[k]*B[i+k];
            }
            A[i]+=p0;
            B[i]+=1.0;
        }
        printf("%.15lf\n",B[0]/(1-A[0]));
    }
    return 0;
}