题意:一个R*C的布局,每个格子中给出停留在原地,往右走一个,往下走一格的概率,起点在(1,1),终点在(R,C),每走一格消耗2的能量,求出最后所需要的能量期望。
思路:dp[i][j]记录从i,j出发到终点的期望,p[i][j][k]记录每格三种状况的概率。
1、dp[ i ][ j ] ---> dp[ i ][ j ],转移概率为p1;
2、dp[ i ][ j ] ---> dp[ i ][ j + 1 ] ,转移概率为p2;
3、dp[ i ][ j ] ---> dp[ i + 1 ][ j ] ,转移概率为p3;
总期望为子期望的加权和,所以得到:
dp[ i ][ j ]= p1 * ( dp[ i ][ j ] + 2 ) + p2 * ( dp[ i ][ j + 1 ] + 2 ) + p3 * ( dp[ i + 1 ][ j ] + 2) ;
化简得:
dp[ i ][ j ]=( 2 + p2 * dp[ i ][ j + 1 ] + p3 * dp[ i + 1 ][ j ] ) / ( 1 - p1) ;
特别注意:可能存在p1=1.0的时候,即说明该点不被访问
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
double dp[1010][1010];
double p[1010][1010][5];
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
memset(dp,0,sizeof(dp));
//memset(p,0,sizeof(p));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
for(int k=0;k<3;k++)
{
scanf("%lf",&p[i][j][k]);
// printf(">>>\n");
}
}
}
for(int i=n;i>0;i--)
{
for(int j=m;j>0;j--)
{
if(p[i][j][0]==1)
continue;
if(i==n&&j==m)
continue;
dp[i][j]=(p[i][j][1]*dp[i][j+1]+p[i][j][2]*dp[i+1][j]+2)/(1.00-p[i][j][0]);
}
}
printf("%.3f\n",dp[1][1]);
}
return 0;
}