关键词:期望dp,待定系数法
题意:已知有三个骰子,分别有k1,k2,k3个面。一开始cnt=0.。当骰子掷为a,b,c时,cnt归零,否则将三个数的和加入cnt。求第一次cnt>n的时,掷筛子的期望次数。
dp[i]:cnt=i时离cnt>n状态还需要掷的期望次数
dp[i]=(dp[0]+∑k1x=1∑k2y=1∑k3z=1(x,y,z)!=(a,b,c)dp[i+x+y+z])+1
式中含dp[0],如何去除dp[0](最终要求的状态——待定系数法
设通式dp[i]=A[i]*dp[0]+B[i],代入上面的递推式,对应各项可分别得到A[i]和B[i]的递推式(不含dp[0])
求出A[0],B[0],dp[0]=A[0]*dp[0]+B[0],求出dp[0]即可!
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<vector>
#include<map>
#define ll long long
using namespace std;
const int maxn = 500+10;
int t;
int n,k1,k2,k3,a,b,c;
double dp[maxn],A[maxn],B[maxn];
int main(){
scanf("%d",&t);
while(t--){
memset(dp,0,sizeof(dp));
memset(A,0,sizeof(A)),memset(B,0,sizeof(B));
scanf("%d%d%d%d%d%d%d",&n,&k1,&k2,&k3,&a,&b,&c);
for(int i=n;i>=0;i--){
double tmp1=0.0,tmp2=0.0;
for(int x=1;x<=k1;x++){
for(int y=1;y<=k2;y++){
for(int z=1;z<=k3;z++){
if(x==a&&y==b&&z==c) continue;
tmp1+=A[i+x+y+z];
tmp2+=B[i+x+y+z];
}
}
}
A[i]=(double)(1+tmp1)/(double)(k1*k2*k3);
B[i]=(double)(1+(tmp2/(double)(k1*k2*k3)));
}
printf("%.10lf\n",B[0]/(1-A[0]));
}
return 0;
}