poj 1322 概率dp

题意：

c种巧克力无限个，取n个后桌面上剩下m个概率是多少。（如果桌面上出现2个一样的巧克力就把它们吃掉）

分析：

我们令dp【i】【j】为取i个巧克力后还剩下j个巧克力

那么dp【i】【j】=dp【i-】【j-1】*（c-j+1）/c+dp【i-1】【j+1】*（j+1）/c

这题有2个坑点

一：n可以取很大，但是n到达一定数量级后概率是收敛的所以不用取那么大

二：能奇偶剪枝：取奇次肯定不会留下偶数个巧克力

ACcode：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#define maxn 1000007
#define inf 0x3f3f3f3f3f
#include <cstring>
using namespace std;
double dp[2][maxn];
int main(){
    int c,n,m;
    while(scanf("%d",&c)&&c){
        scanf("%d%d",&n,&m);
        if(m>c||m>n||(m+n)%2){
            printf("0.000\n");
            continue;
        }
        memset(dp,0,sizeof(dp));
        n=n>1000?1000+n%2:n;
        dp[0][0]=1.0;
        for(int i=1;i<=n;++i)
            for(int j=0;j<=i&&j<=c;++j){
                dp[i%2][j]=0.0;
                if((i+j)%2)continue;
                if(j>0)dp[i%2][j]+=dp[(i+1)%2][j-1]*(double(c-j+1.0)/c);
                if(j+1<=i-1)dp[i%2][j]+=dp[(i+1)%2][j+1]*(double(j+1.0)/c);
            }
        printf("%.3f\n",dp[n%2][m]);
    }
    return 0;
}