欢迎访问LeetCode题解合集
题目描述
一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下
,则骑士的初始健康点数至少为 7。
-2 (K) | -3 | 3 |
---|---|---|
-5 | -10 | 1 |
10 | 30 | -5 (P) |
说明:
-
骑士的健康点数没有上限。
-
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间
题解:
动态规划。
设 f[i][j]
表示骑士从 (i, j)
走到右下角应该具备的最低血量。
但是这题有一个比较有意思的地方:如果我们选择 自顶向下 的方式,即从左上角出发,在前进过程中,每一个位置的最小血量是由后面的路径决定的,因此,自顶向下 方法是不行的。而右下角的最低血量是可以确定的,可以从右下角出发,自底向上。
时间复杂度:$O(n*m)$
额外空间复杂度:$O(n * m)$
class Solution {
public:
int calculateMinimumHP(vector<vector<int>>& dungeon) {
int n = dungeon.size();
int m = dungeon[0].size();
vector<vector<int>> f(n, vector<int>(m));
f[n - 1][m - 1] = dungeon[n - 1][m - 1] < 0 ? -dungeon[n - 1][m - 1] + 1 : 1;
for ( int j = m - 2; j >= 0; --j )
f[n - 1][j] = max( f[n - 1][j + 1] - dungeon[n - 1][j], 1 );
for ( int i = n - 2; i >= 0; --i )
f[i][m - 1] = max( f[i + 1][m - 1] - dungeon[i][m - 1], 1 );
int l, r;
for ( int i = n - 2; i >= 0; --i ) {
for ( int j = m - 2; j >= 0; --j ) {
l = max( f[i + 1][j] - dungeon[i][j], 1 );
r = max( f[i][j + 1] - dungeon[i][j], 1 );
f[i][j] = min( l, r );
}
}
return f[0][0];
}
};
/*
时间:4ms,击败:98.67%
内存:8.7MB,击败:48.91%
*/
可以使用滚动数组优化:
class Solution {
public:
int calculateMinimumHP(vector<vector<int>>& dungeon) {
int n = dungeon.size();
int m = dungeon[0].size();
vector<int> f(m);
f[m - 1] = dungeon[n - 1][m - 1] < 0 ? -dungeon[n - 1][m - 1] + 1 : 1;
for ( int j = m - 2; j >= 0; --j )
f[j] = max( f[j + 1] - dungeon[n - 1][j], 1 );
int l, r;
for ( int i = n - 2; i >= 0; --i ) {
f[m - 1] = max( f[m - 1] - dungeon[i][m - 1], 1 );
for ( int j = m - 2; j >= 0; --j ) {
l = max( f[j] - dungeon[i][j], 1 );
r = max( f[j + 1] - dungeon[i][j], 1 );
f[j] = min( l, r );
}
}
return f[0];
}
};
/*
时间:4ms,击败:98.67%
内存:8.4MB,击败:93.57%
*/