Humora 被困在一个r * c 个小房间组成的矩形迷宫里,起点位位置在(1,1)终点位置在(r, c);每次可以花费两点法力从房间(i, j)以一定的概率跑到(i+1, j)房间 或者(i,j+1)房间或者回到原来的房间。求Humora逃出去的要花费法力的期望。
求期望的概率dp。
设dp[i][j] 表示在房间(i, j)逃出去的花费法力的期望。显然dp[1][1]为题目所求
对于每个房间可能有三种情况分别是移动到房间 (i,j) (i, j+1) (i+1,j) 。可以列出dp方程:
dp[i][j] = p[i][j].a * dp[i][j] + p[i][j].b * dp[i][j+1] + p[i][j].c * dp[i+1][j]; (dp[r][c] = 0.0)
写代码时要注意边界。
还有就是题目说解不小于
1000000 那么说明一定存在解,所以题目数据出现的 p[i][j].a = 1的情况要排除掉。
#include <cmath> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; struct Node { double a, b, c; Node(){} Node(double _a, double _b, double _c) :a(_a),b(_b),c(_c){} }; int r, c; Node p[1010][1010]; double dp[1010][1010]; double DP(int x, int y) { if (dp[x][y] != -1.0) return dp[x][y]; if (x == r && y == c) return 0.0; if (abs(1 - p[x][y].a) < 1e-8) return dp[x][y] = 1.0; double res = 0.0; if (y != c) res += p[x][y].b * DP(x, y+1); if (x != r) res += p[x][y].c * DP(x+1, y); return dp[x][y] = (res + 2.0) * 1.0 / (1.0 - p[x][y].a); } int main () { while (cin >> r >> c) { double _a, _b, _c; for (int i=1; i<=r; i++) { for (int j=1; j<=c; j++) { scanf("%lf %lf %lf", &_a, &_b, &_c); p[i][j] = Node(_a, _b, _c); dp[i][j] = -1.0; } } printf ("%.3lf\n", DP(1, 1)); } return 0; }
版权声明:本文为博主原创文章,未经博主允许不得转载。
时间: 2024-10-11 19:44:23