一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2 (K) -3 3
-5 -10 1
10 30 -5 (P)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/dungeon-game
解题思路
采用动态规划,dp[i][j]
代表从(i, j)
到终点所需最少生命值
从终点往回倒推,dp[i][j] = Max(Min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j], 1)
当前点到终点所需最少HP
= 当前点的右侧和下侧中所需较少的HP
- 当前点提供的HP
如果上述操作使得当前点到终点所需最少HP
为负数,则说明当前点不需要"增加"HP
也可以到达终点,所以设置为1
代码
class Solution {
public int calculateMinimumHP(int[][] dungeon) {
int m = dungeon.length, n = dungeon[0].length;
// dp[i][j]: 从(i, j)到终点所需最少生命值
int[][] dp = new int[m + 1][n + 1];
for (int i = 0; i <= m; i++) {
Arrays.fill(dp[i], Integer.MAX_VALUE);
}
dp[m][n - 1] = dp[m - 1][n] = 1;
// dp[i][j] = Max(Min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j], 1)
for (int i = m - 1; i >= 0; i--) {
for (int j = n - 1; j >= 0; j--) {
dp[i][j] = Math.max(Math.min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j], 1);
}
}
return dp[0][0];
}
}