题意:同时摇三个色子,三个色子的面数分别为 k1 , k2 , k3 ,若摇的结果分别为 a , b, c ; 则积分置零,否则累加积分( i + j + k)(摇色子的结果)。求从零开始摇,积分超过 n 需要摇色子次数的期望。
分析:一开始我真没做出来,推出公式 变了些形 就直接用记忆化搜索写了一遍,结果一直 T , 后来看了别人的题解才明白,这里借鉴一下,
最初的转移方程不难推, 设 dp[ i ] 表示当前积分为 i ,积分超过 n 的期望值 ,pk 为 摇色子得到 k 积分的概率 ,p0为摇色子积分归零的概率
则 ,细看方程已经形成环了 ,一般是用高斯消元解决( 暂时不会( ̄▽ ̄)" ) ,不过
这题有比较巧的做法,这个方程每一项都与dp[ 0 ]有关,且 dp[ 0 ]正是我们要求的结果,所有我们设
, 带入上式整理得:
则
则 A[i] , B[i] 都可以逆推出来,求出A[0] ,B[0] 之后结果就有了!
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N = 600;
double A[N],B[N];
double p[50];
int main()
{
int T;
scanf("%d",&T);
int n,k1,k2,k3,a,b,c;
while(T--)
{
scanf("%d%d%d%d%d%d%d",&n,&k1,&k2,&k3,&a,&b,&c);
double 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;
memset(A,0,sizeof(A));
memset(B,0,sizeof(B));
for(int i=n;i>=0;i--)
{
A[i]=p0,B[i]=1;
for(int k=1;k<=k1+k2+k3;k++)
{
A[i]+=A[i+k]*p[k];
B[i]+=B[i+k]*p[k];
}
}
printf("%.16lf\n",B[0]/(1-A[0]));
}
return 0;
}