分析转自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;
}