2014 Winter Programming School, Kharkiv, day 1 (E. Kapun) D.A fish lunch(概率dp)

题目

飞机盒饭，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;
}