Title

一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。

骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。

有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。

为了尽快到达公主,骑士决定每次只向右或向下移动一步。

编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。

例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。

-2 (K)

-3

3

-5

-10

1

10

30

-5 §

说明:


  • 骑士的健康点数没有上限。
  • 任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。

动态规划

考虑从右下往左上进行动态规划。令 dp[i][j] 表示从坐标 (i,j) 到终点所需的最小初始值。换句话说,当我们到达坐标 (i,j) 时,如果此时我们的路径和不小于 dp[i][j] ,我们就能到达终点。

这样一来,我们就无需担心路径和的问题,只需要关注最小初始值。对于 dp[i][j],我们只要关心 dp[i][j+1] 和 dp[i+1][j] 的最小值 minn。记当前格子的值为 dungeon(i,j),那么在坐标 (i,j) 的初始值只要达到 minn−dungeon(i,j) 即可。同时,初始值还必须大于等于 1。这样我们就可以得到状态转移方程:

d p [ i ] [ j ] = m a x ( m i n ( d p [ i + 1 ] [ j ] , d p [ i ] [ j + 1 ] ) − d u n g e o n ( i , j ) , 1 ) dp[i][j]=max(min(dp[i+1][j],dp[i][j+1])−dungeon(i,j),1) dp[i][j]=max(min(dp[i+1][j],dp[i][j+1])−dungeon(i,j),1)

最终答案即为 dp[0][0]。

边界条件为,当 i=n-1 或者 j=m-1 时,dp[i][j] 转移需要用到的 dp[i][j+1] 和 dp[i+1][j] 中有无效值,因此代码实现中给无效值赋值为极大值。特别地,dp[n-1][m-1] 转移需要用到的 dp[n−1][m] 和 dp[n][m-1] 均为无效值,因此我们给这两个值赋值为 1。

Code

def calculateMinimumHP(self, dungeon: List[List[int]]) -> int:
rows, cols = len(dungeon), len(dungeon[0])
dp = [[10 ** 9] * (cols + 1) for _ in range(rows + 1)]
dp[rows][cols -1] = dp[rows - 1][cols] = 1
for i in range(rows - 1, -1, -1):
for j in range(cols - 1, -1, -1):
minn = min(dp[i + 1][j], dp[i][j + 1])
dp[i][j] = max(minn - dungeon[i][j], 1)
return dp[0][0]

复杂度分析

时间复杂度​:O(N×M),其中 N,M 为给定矩阵的长宽。

空间复杂度​:O(N×M),其中 N,M 为给定矩阵的长宽,注意这里可以利用滚动数组进行优化,优化后空间复杂度可以达到 O(N)。