一些恶魔抓住了公主(P)并将她关在了地下城的右下角。地下城是由 M x N 个房间组成的二维网格。我们英勇的骑士(K)最初被安置在左上角的房间里,他必须穿过地下城并通过对抗恶魔来拯救公主。
骑士的初始健康点数为一个正整数。如果他的健康点数在某一时刻降至 0 或以下,他会立即死亡。
有些房间由恶魔守卫,因此骑士在进入这些房间时会失去健康点数(若房间里的值为负整数,则表示骑士将损失健康点数);其他房间要么是空的(房间里的值为 0),要么包含增加骑士健康点数的魔法球(若房间里的值为正整数,则表示骑士将增加健康点数)。
为了尽快到达公主,骑士决定每次只向右或向下移动一步。
编写一个函数来计算确保骑士能够拯救到公主所需的最低初始健康点数。
例如,考虑到如下布局的地下城,如果骑士遵循最佳路径 右 -> 右 -> 下 -> 下,则骑士的初始健康点数至少为 7。
-2 (K) -3 3
-5 -10 1
10 30 -5 (P)
说明:
骑士的健康点数没有上限。
任何房间都可能对骑士的健康点数造成威胁,也可能增加骑士的健康点数,包括骑士进入的左上角房间以及公主被监禁的右下角房间。
思路:
动态规划,dp[i][j]代表行进到(i,j)时候剩余的健康点数,若要行进到下一个点,则要加上当前点dungeon[i][j]。因为只能往右或者往下移动,所以dp[i][j]=min(dp[i][j+1]-dungeon[i][j],dp[i+1][j]-dungeon[i][j])。此外,还要考虑边界条件的限制,在最下面和最右边时候min只有一项,还有就是dp[i][j]的最小值是1,需要加上对负数的判断。具体实现代码如下。
class Solution {
public:
int calculateMinimumHP(vector<vector<int>>& dungeon) {
if(!dungeon.size() || !dungeon[0].size())
{
return 0;
}
int rows=dungeon.size();
int cols=dungeon[0].size();
vector<vector<int>> dp;
dp=vector<vector<int>>(rows,vector<int>(cols,0));
for(int i=rows-1;i>=0;i--)
{
for(int j=cols-1;j>=0;j--)
{
if(i==rows-1 && j==cols-1)
{
dp[i][j]=max(1,1-dungeon[i][j]);
}
else if(i==rows-1)
{
dp[i][j]=max(dp[i][j+1]-dungeon[i][j],1);
}
else if(j==cols-1)
{
dp[i][j]=max(dp[i+1][j]-dungeon[i][j],1);
}
else
{
int h1=max(dp[i][j+1]-dungeon[i][j],1);
int h2=max(dp[i+1][j]-dungeon[i][j],1);
dp[i][j]=min(h1,h2);
}
}
}
return dp[0][0];
}
};
网友评论