一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2(K) | -3 | 3 |
---|---|---|
-5 | -10 | 1 |
10 | 30 | -5(P) |
说明:
- 骑士的健康点数没有上限。
- 任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间
解题思路:
右下的格子与其上方和左侧的格子状态相关,经过转移得来,我们自然联想到使用动态规划方法。这里我们选择从右下到左上来动态规划,原因在于如果按照从左上到右下的方式,所需要的最小生命点数无法更新。按照从右下到左上的方式,列出状态转移方程:
最终答案为dp[0] [0],代码按照从右下到左上得到dp的方式编写:
class Solution:
def calculateMinimumHP(self, dungeon: List[List[int]]) -> int:
x = len(dungeon)
y = len(dungeon[0])
dp = [[None for __ in range(y)] for __ in range(x)]
# 确定右下角
dp[-1][-1] = 1 if dungeon[-1][-1] >= 0 else 1 - dungeon[-1][-1]
# 确定最右边一整列
for i in range(x-2, -1, -1):
tmp = dp[i+1][-1]-dungeon[i][-1]
dp[i][-1] = 1 if tmp <= 0 else tmp
# 确定最下方一整行
for i in range(y-2, -1, -1):
tmp = dp[-1][i+1]-dungeon[-1][i]
dp[-1][i] = 1 if tmp <= 0 else tmp
# 动态规划其余的dp列表元素
for i in range(x-2, -1, -1):
for j in range(y-2, -1, -1):
tmp = min(dp[i][j+1], dp[i+1][j])-dungeon[i][j]
dp[i][j] = 1 if tmp <= 0 else tmp
return dp[0][0]