一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2 (K) -3 3
-5 -10 1
10 30 -5 §
代码
class Solution(object):
def calculateMinimumHP(self, dungeon):
m = len(dungeon)
n = len(dungeon[0])
ans = [[0 for i in range(n)] for j in range(m)]
for x in range(m - 1, -1, -1):
for y in range(n - 1, -1, -1):
s1 = 0
s2 = 0
if x < m - 1:
s1 = ans[x + 1][y]
if y < n - 1:
s2 = ans[x][y + 1]
if s1 == 1 or s2 == 1 or (s1 == 0 and s2 == 0):
if dungeon[x][y] >= 0:
ans[x][y] = 1
else:
ans[x][y] = 1 - dungeon[x][y]
elif s1==0:
ans[x][y] = max(1, s2 - dungeon[x][y])
elif s2==0:
ans[x][y] = max(1, s1 - dungeon[x][y])
else:
ans[x][y] = max(1, min(s1, s2) - dungeon[x][y])
return ans[0][0]
思路就是倒过来走,到达最后一个位置时候需要多少能量,那他的上一格能量就能得知
比如如何得知x,y的能量呢?判断当前格子是补充还是扣除,在去判断他的下一条路径所需能量,获取当前格子的能量最小值