持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第25天,点击查看活动详情
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
说明:
-
骑士的健康点数没有上限。
-
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
思路
本题我们可以用动态规划算法求解。用fn(i,j) (0 <= i < M,0 <= j < N)表示在第i行j列房间到右下角房间消耗的体力值,由于骑士只能往右或者往下走,骑士应该选择这两天道路中消耗健康点最小的路径前往公主的房间。又因为骑士在任意房间的健康值都应该大于0,如果两条路径中最小消耗值大于等于0,说明后序步骤不需要消耗健康值,fn(i, j) = dungeon[i,j],如果小于0,则fn(i,j) = dungeon[i, j] + Math.max(fn(i + 1 , j), fn(i, j - 1))。初始值fn(m-1,n-1) = dungeon[m-1, n-1],返回值 Math.max(1 - fn(0,0), 1)
解题
/**
* @param {number[][]} dungeon
* @return {number}
*/
var calculateMinimumHP = function (dungeon) {
const m = dungeon.length;
const n = dungeon[0].length;
const dp = new Array(m * n).fill(null);
dp[m * n - 1] = dungeon[m - 1][n - 1];
const dfs = (i, j) => {
let idx = i * n + j;
if (dp[idx] === null) {
let res = Number.MIN_SAFE_INTEGER;
if (i < m - 1) {
res = dfs(i + 1, j);
}
if (res < 0 && j < n - 1) {
res = Math.max(dfs(i, j + 1), res);
}
dp[idx] = res >= 0 ? dungeon[i][j] : dungeon[i][j] + res;
}
return dp[idx];
};
dfs(0, 0);
return dp[0] >= 0 ? 1 : 1 - dp[0];
};