1 题目
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2 (K) -3 3
-5 -10 1
10 30 -5 §
说明:
骑士的健康点数没有上限。
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
2 解析
反过来求,从右下角往左上角求
状态:能到达左上角的最小健康值,表示为dp[i][j]
状态转移:
记当前格子的值为 dungeon(i,j),那么在坐标 (i,j) 的初始值只要达到 minn−dungeon(i,j) 即可。同时,初始值还必须大于等于 1。这样我们就可以得到状态转移方程:
$dp[i][j]=max(min(dp[i+1][j],dp[i][j+1])−dungeon(i,j),1)$
最终答案即为 dp[0][0]
注意:初始化n+1行,n+1列的矩阵。除了在公主位置的左一个位置和上一个位置,初始值是1,其他位置初始值为无限大。
3 Python实现
class Solution:
def calculateMinimumHP(self, dungeon: List[List[int]]) -> int:
m = len(dungeon)
n = len(dungeon[0])
# 注意,要初始化n+1的矩阵
dp = [[float('inf')]*(n+1) for _ in range(m+1)]
# 初始化 ,到右下角前,前一步最小值是1
dp[m][n-1] =dp[m-1][n] = 1
for i in range(m-1,-1,-1):
for j in range(n-1,-1,-1):
minn = min(dp[i][j+1],dp[i+1][j])
dp[i][j] = max(minn-dungeon[i][j],1)
return dp[0][0]