概率DP种类问题 UVA - 10648 && CodeForces - 768D

最简单的例子UVA - 10648 

题意：有n块巧克力，m个盒子，巧克力随机放入其中的一个盒子，问至少有一个盒子为空的概率。

这时我们就要求m个盒子全满的概率，答案为1减去这个概率。

当要放一个巧克力时，只有放在已经有巧克力的盒子和放在没有巧克力的盒子两种情况。

所以i个巧克力放在j个盒子的概率就等于i-1个巧克力放在j个盒子的概率加上i-1个巧克力放在j-1个盒子的概率。

所以状态转移方程为 dp[i][j]=dp[i-1][j]*(j*1.0)/m+dp[i-1][j-1]*(m-j+1)/m;

代码：

#include<bits/stdc++.h>

using namespace std;
const int N=205;
double dp[N][N];
int n,m;

int main()
{
    int Case=0;
    while(~scanf("%d",&n))
    {
        Case++;
        memset(dp,0,sizeof(dp));
        if(n==-1)break;
        scanf("%d",&m);
        dp[1][1]=1.0;
        for(int i=2;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                dp[i][j]=dp[i-1][j]*(j*1.0)/m+dp[i-1][j-1]*(m-j+1)/m;
            }
        }
        printf("Case %d:%.7f\n",Case,1-dp[n][m]);
    }
    return 0;
}

还有一个题目是 CodeForces - 768D 

题意：需要k种水晶，每种水晶每次随机出现，求每个水晶都至少出现一次的概率大于等于p/2000需要几天。

和刚才的题目差不多，只需要通过dp打出表，每次查询输出。

dp[i][j]表示i天j种水晶的概率

状态转移方程:  dp[i][j]=dp[i-1][j]*(j*1.0)/n+dp[i-1][j-1]*(n-(j-1))/n;

代码：

#include<bits/stdc++.h>

using namespace std;
const double eps=1e-10;
const int N=10005;
const int M=1005;

double dp[N][M];//第i天得到j种水晶的概率

int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    dp[1][1]=1.0;
    for(int i=2;i<N;i++)
    {
        for(int j=1;j<=n;j++)
        {
            dp[i][j]=dp[i-1][j]*(j*1.0)/n+dp[i-1][j-1]*(n-(j-1))/n;
        }
    }
    while(q--)
    {
        double p;
        scanf("%lf",&p);
        for(int i=0;i<=N;i++)
        {
            if(dp[i][n]-p/2000>=eps)
            {
                printf("%d\n",i);
                break;
            }
        }
    }
    return 0;
}