跳转至

LeetCode: 174. 地下城游戏

1、题目描述

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

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

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

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

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

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

-2(K) -3 3
-5 -10 1
10 30 -5(P)

说明:

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

2、解题思路

  • 动态规划
  • 由于骑士只能向右或向下,因此我们从右下角开始更新
dp[i][j] : 表示当前节点到右下角的节点需要的最少的健康点

dp[i][j] = max(1, min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j])
在右方和下方中找出最少血液的,并且判断当前房间是多少健康点,至少提供一点健康点
因此,通过减的方式,如果结果小于0,表示当前房间是提供健康点的,更新为1即可
如果结果大于0,表示当前房间消耗健康点
class Solution:
    def calculateMinimumHP(self, dungeon: List[List[int]]) -> int:
        row, col = len(dungeon), len(dungeon[0])

        dp = [[0 for _ in range(col)] for _ in range(row)]

        dp[row - 1][col - 1] = max(1, 1 - dungeon[row-1][col - 1])

        for i in range(row - 2, -1, -1):
            dp[i][col - 1] = max(1, dp[i + 1][col - 1] - dungeon[i][col - 1])

        for j in range(col - 2, -1, -1):
            dp[row - 1][j] = max(1, dp[row - 1][j + 1] - dungeon[row - 1][j])

        for i in range(row - 2, -1, -1):
            for j in range(col - 2, -1, -1):
                dp[i][j] = max(1, min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j])

        return dp[0][0]