BNU29034 求期望的dp

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

求概率的dp。

代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<string>
#include<queue>
#include<cmath>
#include<algorithm>
#include<cstring>
#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;
}