题目链接:https://leetcode-cn.com/problems/dungeon-game/
难度:困难
描述:
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N
个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2 (K) | -3 | 3 |
---|---|---|
-5 | -10 | 1 |
10 | 30 | -5 (P) |
题解
从右下到左上动态规划。
class Solution:
def calculateMinimumHP(self, dungeon: List[List[int]]) -> int:
m, n = len(dungeon), len(dungeon[0])
# r[i][j]表示从当前位置到公主处所需的最少血量
r = [[float("inf")] * (n+1) for _ in range(m+1)]
r[m-1][n] = 1
r[m][n-1] = 1
for i in range(m-1, -1, -1):
for j in range(n-1, -1, -1):
# 最小血量不能小于1
r[i][j] = max(min(r[i+1][j], r[i][j+1]) - dungeon[i][j], 1)
return r[0][0]