HDU 3853 期望概率DP

期望概率DP简单题

从[1,1]点走到[r,c]点，每走一步的代价为2

给出每一个点走相邻位置的概率，共3中方向，不动： [x,y]->[x][y]=p[x][y][0] ，  右移：[x][y]->[x][y+1]=p[x][y][1];  左移：[x][y]->[x+1][y]=p[x][y][2];

问最后走到[r,c]的期望

dp[i][j]为从[i][j]点走到[r][c]的期望

有方程：

dp[i][j]=    (dp[i][j]+2)*p[i][j][0]  +   (dp[i][j+1]+2)*p[i][j][1]    +    (dp[i+1][j]+2)*p[i][j][2] ;

移项合并： dp[i][j]= (  (dp[i][j+1]+2)*p[i][j][1]    +    (dp[i+1][j]+2)*p[i][j][2]   +   p[i][j][0]*2  )  /  (1-p[i][j][0])

特判p[i][j][0]==1 的情况

#include "stdio.h"
#include "string.h"
#include "math.h"
double p[1010][1010][3],dp[1010][1010];

int main()
{
    int n,m,i,j;

    while (scanf("%d%d",&n,&m)!=EOF)
    {
        for (i=1;i<=n;i++)
            for (j=1;j<=m;j++)
            scanf("%lf%lf%lf",&p[i][j][0],&p[i][j][1],&p[i][j][2]);

        memset(dp,0,sizeof(dp));

        for (i=n;i>=1;i--)
            for (j=m;j>=1;j--)
            {
                dp[i][j]=(dp[i][j+1]+2)*p[i][j][1]+(dp[i+1][j]+2)*(p[i][j][2])+p[i][j][0]*2;
                if (fabs(p[i][j][0]-1)<=0.00000000001) dp[i][j]=0;
                else dp[i][j]/=(1-p[i][j][0]);
            }
        printf("%.3lf\n",dp[1][1]);
    }
    return 0;
}