题目
飞机盒饭,n1(0<=n1<=1e3)包肉,n2(0<=n2<=1e3)包鱼,共n(1<=n<=n1+n2)个人
Alex在第I位(1<=I<=n),前面的人每个人有P%的概率选择鱼(0<=P<=100)
但是,当没有鱼或没有肉的时候,前面的人将被迫选择,
求Alex能选到鱼的概率
题解
dp[i][j]:代表当前还剩i包肉j包鱼的概率,剩的下标小了就代表被选了
当I选时,前面已选I-1个,则i+j总数应为n1+n2-(I-1),且合法的鱼的数量应为[1,n2]
注意被迫选择的情况*概率1,其余情况对应*p和*(1-p)的转移
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e3+5;
double dp[N][N],p,ans;//dp[i][j]:当前还剩i顿肉j顿鱼的概率
int n,n1,n2,I,P;
int main()
{
scanf("%d%d%d%d%d",&n,&n1,&n2,&I,&P);
p=P/100.0;
for(int i=n1;i>=0;--i)
{
for(int j=n2;j>=0;--j)
{
if(i==n1&&j==n2){dp[i][j]=1;continue;}
if(i==0&&j==0){dp[i][j]=dp[i+1][j]+dp[i][j+1];continue;}
if(i==0){dp[i][j]=dp[i+1][j]*(1-p)+dp[i][j+1];continue;}
if(j==0){dp[i][j]=dp[i+1][j]+dp[i][j+1]*p;continue;}
dp[i][j]=dp[i+1][j]*(1-p)+dp[i][j+1]*p;
}
}
for(int i=n1;i>=0;--i)//i顿肉
{
int j=n1+n2-(I-1)-i;
if(j>n2||j<=0)continue;//鱼在合法范围内
ans+=dp[i][j];
}
printf("%.8lf\n",ans);
return 0;
}