睡前小dp-hdu3853-概率dp

http://acm.hdu.edu.cn/showproblem.php?pid=3853

膜猴烧酒madoka

讲定义为dp[i][j] 位置为ij的魔法值期望，可以发现dp[i][j] = dp[i][j]*p1[i][j]+dp[i][j+1]*p2[i][j]+dp[i+1][j]*p3[i][j]+2

最后可以表示为

dp[x][y] = 1.0/(1-p[x][y][0])*(p[x][y][1]*dp[x][y+1]+p[x][y][2]*dp[x+1][y]+2.0);

递推的时候从右下角开始，向左上角递推。

有一个坑是p1=1 时，会出错，所以要特判一下。直接跳过。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;

int R,C;
double dp[1010][1010],p[1010][1010][3];

int main()
{
    while(~scanf("%d%d",&R,&C))
    {
        for(int i=1;i<=R;i++)
        {
            for(int j=1;j<=C;j++)
            {
                scanf("%lf%lf%lf",&p[i][j][0],&p[i][j][1],&p[i][j][2]);
            }
        }

        memset(dp,0,sizeof dp);

        for(int x=R;x>=1;x--)
        {
            for(int y=C;y>=1;y--)
            {
                if(x==R&&y==C) continue;
                if(abs(p[x][y][0]-1.0)<(1e-7)) continue;
                dp[x][y] = 1.0/(1-p[x][y][0])*(p[x][y][1]*dp[x][y+1]+p[x][y][2]*dp[x+1][y]+2.0);
            }
        }

    /*
            for(int i=1;i<=R;i++)
        {
            for(int j=1;j<=C;j++)
            {
                printf("%lf ",dp[i][j]);
            }
            printf("\n");
        }
    */
        printf("%.3lf\n",dp[1][1]);
    }
}