ZOJ 3582 Back to the Past 概率DP

题意：一开始脑残吧组合数公式算错了，要先把c[i][i]和c[i][0]赋1才行。

又是条件概率，要计算灯亮的概率是，灯的位置也必须考虑在内的

#include<cstdio>
#include<cstring>
const int LMT=55;
double dp[LMT][LMT],p,myp[LMT],nmyp[LMT],c[LMT][LMT];
//考虑位置啊。。。
void init(void)
{
	int i,j;
	memset(dp,0,sizeof(dp));
	c[0][0]=nmyp[0]=myp[0]=1.0;
	for(i=1;i<LMT;i++)
	{
		myp[i]=myp[i-1]*p;
		nmyp[i]=nmyp[i-1]*(1-p);
	}
	for(i=0;i<LMT;i++)c[i][i]=c[i][0]=1.0;
	for(i=1;i<LMT;i++)
		for(j=1;j<i;j++)
			c[i][j]=c[i-1][j-1]+c[i-1][j];
}
int main(void)
{
	int n,m,i,j,ii,jj;
	while(~scanf("%d%d%lf",&n,&m,&p)&&(m+n+p))
	{
		init();
		for(i=n;i>=0;i--)
			for(j=n;j>=0;j--)
			{
				if(i<m||j<m)dp[i][j]=(dp[i][j]+1)/(1-nmyp[n-i]*nmyp[n-j]);
				for(ii=0;ii<=i;ii++)
					for(jj=0;jj<=j;jj++)
						if(ii!=i||jj!=j)
						dp[ii][jj]+=myp[i-ii]*nmyp[n-i]*myp[j-jj]*nmyp[n-j]*c[n-ii][i-ii]*c[n-jj][j-jj]*dp[i][j];
			}
			printf("%.6lf\n",dp[0][0]);
	}
	return 0;
}