睡前小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/(-p[x][y][])*(p[x][y][]*dp[x][y+]+p[x][y][]*dp[x+][y]+2.0);

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

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

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

using namespace std;

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

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

        memset(dp,,sizeof dp);

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