LeetCode 174
这个题目使用动态规划,从数组右下角开始向左上角解决会比较容易。
使用一个二位数组来记录在每个位置上需要的生命值,最小是1(如果是0,那么就挂了)。为了方便处理,我们多给这个二维数组加一条边(m+1 * n+1)。在找到公主之后剩下的生命值最小是1.
每次我们可以从左边过来,或者上边过来,反过来算的时候,那么即使从右边或者下边过来小的这个,min(dp【i+1】【j】, dp【i】【j+1】),这些是他们需要进入下一个的生命值,对于当前的伤害或者加强,也是需要减掉的,所以这个min(dp【i+1】【j】, dp【i】【j+1】)= dp【i】【j】+dungeon【i】【j】,转换一下 dp【i】【j】 = min(dp【i+1】【j】, dp【i】【j+1】)-dungeon【i】【j】(注意我们还有一个条件,就是不能死,所以dp【i】【j】最小是1)。把公式转为代码:
class Solution:
def calculateMinimumHP(self, dungeon: List[List[int]]) -> int:
m, n = len(dungeon), len(dungeon[0])
dp = [[sys.maxsize]*(n+1) for _ in range(m+1)]
dp[m-1][n], dp[m][n-1] = 1, 1
for i in range(m-1,-1,-1):
for j in range(n-1,-1,-1):
dp[i][j] = max(min(dp[i+1][j],dp[i][j+1])-dungeon[i][j],1)
return dp[0][0]