原题地址:https://leetcode-cn.com/problems/dungeon-game/description/
题目描述:
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2 (K) -3 3
-5 -10 1
10 30 -5 (P)
说明:
骑士的健康点数没有上限。
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
解题方案:
很明显,属于dp的题型。dp是二维数组,开始将起始点作为dp的初始值,后来发现这种计算方式过于复杂,好像也不合逻辑。遇到麻烦难解决的问题,需要变通一下,将终点作为dp的初始值,最后返回的是dp[0][0]。
参考地址:https://blog.csdn.net/zxzxzx0119/article/details/81946590
代码:
class Solution {
public:
int calculateMinimumHP(vector<vector<int>>& dungeon) {
int n = dungeon.size(), m = dungeon[0].size();
if(n <= 0 || m <= 0) return 0;
vector<vector<int>> dp(n, vector<int>(m, 0));
dp[n - 1][m - 1] = dungeon[n - 1][m - 1] > 0 ? 1: - dungeon[n-1][m-1] + 1;
for(int j = m - 2; j >= 0; j --)
dp[n - 1][j] = dungeon[n - 1][j] >= dp[n - 1][j + 1] ? 1 : dp[n - 1][j + 1] - dungeon[n -1 ][j];
for(int i = n - 2; i >= 0; i --)
dp[i][m - 1] = dungeon[i][m - 1] >= dp[i + 1][m - 1] ? 1 : dp[i + 1][m - 1] - dungeon[i][m - 1];
for(int i = n - 2; i >= 0; i --)
for(int j = m - 2; j >= 0; j --)
dp[i][j] = (dungeon[i][j] >= min(dp[i][j + 1], dp[i + 1][j])) ? 1 : min(dp[i][j + 1], dp[i + 1][j]) - dungeon[i][j];
return dp[0][0];
}
};