一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下
,则骑士的初始健康点数至少为 7。
-2 (K) | -3 | 3 |
-5 | -10 | 1 |
10 | 30 | -5 (P) |
说明:
-
骑士的健康点数没有上限。
- 任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
解题: 本题是一个动态规划问题,需要从终点往回推,计算进入每一格需要的最低血量,定义dp[n][m]数组存储进入每个房间的最低血量,即 dp[i][j]=min(dp[i+1][j] , dp[i][j+1]) -dungeon[i][j],同时dp[i][j]>=1。
因为最后一行与最后一列只需要考虑右边房间或下面房间,因此先单独计算dp[n-1][j]和dp[i][m-1]:
dp[0][m-1] | ||
dp[1][m-1] | ||
dp[n-1][0] | dp[n-1][1] | max(-(dungeon[n-1][m-1]-1),1) |
接下来利用上式计算dp[i][j]=min(dp[i+1][j] , dp[i][j+1]) -dungeon[i][j],同时dp[i][j]>=1。最终dp[0][0]为需要的最少血量。
Python代码如下:
class Solution:
def calculateMinimumHP(self, dungeon):
"""
:type dungeon: List[List[int]]
:rtype: int
"""
n=len(dungeon)
m=len(dungeon[0])
dp=[[0 for i in range(m)] for i in range(n)]
dp[n-1][m-1]=max(-(dungeon[n-1][m-1]-1),1)
for i in range(m-1):
dp[n-1][m-2-i]=max(dp[n-1][m-1-i]-dungeon[n-1][m-2-i],1)
for i in range(n-1):
dp[n-2-i][m-1]=max(dp[n-1-i][m-1]-dungeon[n-2-i][m-1],1)
for i in range(n-1):
for j in range(m-1):
tempval=min(dp[n-1-i][m-2-j],dp[n-2-i][m-1-j])-dungeon[n-2-i][m-2-j]
dp[n-2-i][m-2-j]=max(1,tempval)
return dp[0][0]