问题描述
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2(k) | -3 | 3 |
---|---|---|
-5 | -10 | 1 |
10 | 30 | -5§ |
说明:
骑士的健康点数没有上限。
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
思路
(1)DFS搜索,每次只能往左或者往右。不过很容易就超时了。
(2)DP算法。
有两个思路,一个是正向DP,一个是方向DP
1、正向DP
起始的最低HP min_hp = 1,最高需要的HP max_hp 是所有负值相加,取反加1(因为最差情况下就是把所有的负值得点都经过了)。
因此这时候可以使用二分搜索+正向DP的解法。
假设当前所需要的HP是mid_hp = (min_hp + max_hp) / 2;
如果mid_hp血量的时候只要能够到达终点,则max_hp = mid_hp
如果mid_hp血量的时候不能达到终点,则min_hp = mid_hp + 1
2、反向DP
令dp[i][j] 为(i,j)这个点到底终点所需要的最低HP,
则dp[i][j] = min(dp[i+1][j], dp[i][j+1]) - dungeon[i][j], 如果dp[i][j]小于或等于0,则说明(i,j)这个点到终点可达,只需要dp[i][j] = 1即可。
代码
正向DP的代码
class Solution {
public:
int calculateMinimumHP(vector<vector<int>>& dungeon) {
int ans = 1e9;
int max_need_hp = 0;
int min_need_hp = 1;
for (int i = 0;i < dungeon.size(); i++)
{
for(int j = 0; j < dungeon[0].size(); j++)
{
if (dungeon[i][j] < 0)
max_need_hp += -dungeon[i][j];
}
}
max_need_hp += 1;
while(min_need_hp < max_need_hp)
{
int mid = (min_need_hp + max_need_hp) >> 1;
if (canRearchEnd(dungeon, mid))
max_need_hp = mid;
else
min_need_hp = mid + 1;
}
return max_need_hp;
}
bool canRearchEnd(vector<vector<int>>& dungeon, int hp)
{
int m = dungeon.size();
int n = dungeon[0].size();
vector<vector<int>> dp(m, vector<int>(n,-1e9));
for (int i = 0;i < m; i++)
{
for (int j = 0; j < n; j++)
{
if (i == 0 && j == 0)
{
dp[i][j] = hp + dungeon[i][j];
}
else if (i == 0)
{
dp[i][j] = dp[i][j-1] + dungeon[i][j];
}
else if (j == 0)
{
dp[i][j] = dp[i-1][j] + dungeon[i][j];
}
else
{
dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + dungeon[i][j];
}
if (dp[i][j] <= 0)
dp[i][j] = -1e9;
}
}
return dp[m-1][n-1] > 0;
}
};