poj 1322 Chocolate

//poj 1322 Chocolate DP/*一个很显然的事实是，在某时刻，桌子上任意两根头发的颜色一定都不相同。令f[i][j]表示i次操作后，桌子上有j根头发的概率。对于每个f[i][j]：若新拔下的头发与目前桌子上的头发颜色都不相同（概率为(c-j)/c），那么f[i+1][j+1]+=f[i][j]*(c-j)/c。若新拔下的头发与目前桌子上的某根头发颜色相同（概率为j/c），那么f[i+1][j-1]+=f[i][j]*j/c。初始化f[0][0]=1，f[0][i]=0，若M>C，则答案为0，否则答案为f[N][M]。但是这样显然要TLE。由于题目要求的精度比较低，而且N很大时答案趋于稳定，所以，若N很大且N为奇数，则把N赋值为1001，若N很大且N为偶数，则把N赋值为1000，然后暴力转移就可以。*/#include <iostream>#include <algorithm>using namespace std;const int inf = 1<<28;#define eps 0.000001int n,m,c;double dp[1005][1005];int main(){ while (scanf("%d",&c) && c!=0) { scanf("%d%d",&n,&m); if (m>c) { printf("0.000/n"); continue; } if (n>1001) //运用n足够大时结果的稳定性进行压缩，但要保留奇偶性 { if (n%2) n=1001; else n=1000; } memset(dp,0,sizeof(dp)); dp[0][0]=1; for (int i=1;i<=n;i++) { for (int j=i;j>=0;j--) { if (j && dp[i-1][j-1]>eps) dp[i][j]=dp[i-1][j-1]*(c-j+1)/c; //浮点运算容易TLE，加个零判断 if (dp[i-1][j+1]>eps) dp[i][j]+=dp[i-1][j+1]*(j+1)/c; // printf("%lf ",dp[i][j]); }// printf("/n"); } printf("%.3lf/n",dp[n][m]); } system("pause"); return 0;}