BNU29034 求期望的dp

http://www.bnuoj.com/bnuoj/problem_show.php?pid=29034

求概率的dp。

代码:

#include
#include
#include
#include
#include
#include
#include
#include
#define maxn 1005
#define ll long long
using namespace std;
double dp[maxn][maxn];
int num,n;
double L,R;
double DP()
{
    memset(dp,0,sizeof(dp));
    dp[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=n;j++)
        {
            dp[i][j]=dp[i][j]+L*dp[i-1][j+1]+dp[i-1][j]*(1.0-L-R);
            if(j>=1)
            dp[i][j]+=R*dp[i-1][j-1];
            else
            dp[i][j]+=L*dp[i-1][j];
        }
    }
    double ret=0;
    for(int i=0;i<=n;i++)
    {
        ret=ret+dp[n][i]*i;
    }
    return ret;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&num,&n);
        scanf("%lf%lf",&L,&R);
        printf("%d %.4lf\n",num,DP());
    }
    return 0;
}


你可能感兴趣的:(动态规划)