Leetcode-174.地下城游戏
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2(K) -3 3
-5 -10 1
10 30 -5§
说明:
骑士的健康点数没有上限。
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
思路:
动态规划问题,声明一个与dungeon相同维度的二维数组res,res第i行j列表示到该位置最少所需的血量,然后从右下角向左上角依次循环遍历数组。遍历之前,先计算完成最左边和最下面的res情况,然后再遍历剩余的res位置。
代码:
class Solution {
public:
int calculateMinimumHP(vector<vector<int>>& dungeon) {
vector<vector<int> > res(dungeon.size(), vector<int>(dungeon[0].size(), 0));
int row = dungeon.size();
int col = dungeon[0].size();
if(row == 0){
return 0;
}
res[row - 1][col - 1] = max(1, 1 - dungeon[row - 1][col - 1]);
for(int i = col - 2; i >= 0; i--){
res[row - 1][i] = max(1, res[row - 1][i + 1] - dungeon[row - 1][i]);
}
for(int i = row - 2; i >= 0; i--){
res[i][col - 1] = max(1, res[i + 1][col - 1] - dungeon[i][col - 1]);
}
for(int i = row -2; i >= 0; i--){
for(int j = col - 2; j >= 0; j--){
int min_ = min(res[i + 1][j], res[i][j + 1]);
res[i][j] = max(1, min_ - dungeon[i][j]);
}
}
return res[0][0];
}
};