zoj 3329 One Person Game(遞推方程轉化求解係數)
阿新 • • 發佈:2019-01-07
題意:有三個骰子,分別有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<math.h> #include<string.h> #include<cstdio> using namespace std; int main() { int i,j,k,f,m,n,a,b,c,k1,k2,k3,t; double A[600],B[600],p[600],pp,res; scanf("%d",&t); while(t--) { scanf("%d%d%d%d%d%d%d",&n,&k1,&k2,&k3,&a,&b,&c); memset(A,0,sizeof(A)); memset(B,0,sizeof(B)); memset(p,0,sizeof(p)); pp=1.0/k1/k2/k3; for(i=1; i<=k1; i++) { for(j=1; j<=k2; j++) { for(k=1; k<=k3; k++) { if(i!=a || j!=b || k!=c) p[i+j+k]+=pp; } } } for(i=n; i>=0; i--) { for(j=3; j<=k1+k2+k3; j++) { A[i]+=A[j+i]*p[j]; B[i]+=B[j+i]*p[j]; } B[i]++; A[i]+=pp; } double res=B[0]/(1-A[0]); printf("%.16lf\n", res); } return 0; }