POJ1322 Chocolate DP

题意：

c种巧克力，每种数量无穷大，取出n个放桌上，如果出现同种巧克力的就必须吃掉，即桌面上同种巧克力只能有0个或者1个。问取出n个后剩余m种巧克力的概率。

思路：

显然不可以直接枚举，需要用dp。

用每次取之后桌面上的巧克力种类数量做状态，容易写出状态转移方程如下：

dp[i][j]=dp[i-1][j-1]*(c-j+1.0)/c+dp[i-1][j+1]*(j+1.0)/c;

因为n非常大，用滚动数组dp[2][N]实现。

刚开始写出如下代码，直接提交，发现TLE。

#include<iostream>
#include<algorithm>
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
using namespace std;
const int N=1000005;
double dp[2][N];
int c,n,m;
int main()
{
	while(scanf("%d%d%d",&c,&n,&m),c!=0)
	{
		memset(dp,0,sizeof(dp));
		
		dp[1][1]=1.0;
		for(int i=2;i<=n;i++)
		{
			dp[i%2][0]=dp[(i-1)%2][1]/c;
			for(int j=1;j<=i&&j<c;j++)
			{
				dp[i%2][j]=dp[(i-1)%2][j-1]*(c-j+1.0)/c+dp[(i-1)%2][j+1]*(j+1.0)/c;
			}
		}

		printf("%.3lf\n",dp[n%2][m]);
	}
	
	return 0;
}

仔细分析：

按照 5 100 2的范例写下来：0代表概率为0，*代表概率大于0，一行6位分别代表m=0,1,2,3,4,5

1---0*0000

2---*0*000

3---0*0*00

4---*0*0*0

5---0*0*0*

6---*0*0*0

...

所以当n为偶数时，m为奇数的肯定为0.即(m+n)%2 直接判为0.

可惜这样还是TLE.

继续找：

因为输出只要求小数点后三位，猜想,当n很大时误差可以忽略不计。猜想n>1000时，直接将n取1000或者1001即可。

此猜想AC了。

从统计学的角度上看，

当n无穷大时，就是样本无穷多，此时概率收敛为一个稳定值也是可以理解的。

具体证明可以看discuss。

#include<iostream>
#include<algorithm>
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
using namespace std;
const int N=105;
double dp[2][N];
int c,n,m;
int main()
{
	while(scanf("%d%d%d",&c,&n,&m),c!=0)
	{
		memset(dp,0,sizeof(dp));
		if(m>c||m>n||(m+n)%2)
		{
			printf("0.000\n");
			continue;
		}
		if(n>1000)
		{
			n=1000+n%2;
		}
		dp[0][0]=1.0;//n=0,m=0时候输出1.000，看了discuss才知道的，否则一直WA。
		dp[1][1]=1.0;
		for(int i=2;i<=n;i++)
		{
			dp[i%2][0]=dp[(i-1)%2][1]/c;
			dp[i%2][c]=dp[(i-1)%2][c-1]/c;
			for(int j=1;j<=i&&j<c;j++)
			{
				dp[i%2][j]=dp[(i-1)%2][j-1]*(c-j+1.0)/c+dp[(i-1)%2][j+1]*(j+1.0)/c;
			}
		}
		printf("%.3lf\n",dp[n%2][m]);
	}
	
	return 0;
}