题目大意: 有一个2D方格,每个方格有个数字,要求从左上角走到右下角,没次走的方向只能是右边和下面,如果格子的数字大于0,表示可以在这个格子加hp,如果小于0就要减掉相应数字的hp.如果到达任何一个格子时,hp为0,表示不能达到这个位置。求刚开始需要初始化多少hp才能保证走到右下角的格子。
解题思路:动态规划。
求从右下角到左上角需要的hp值。
dp[i][j]表示从(i,j)这个位置到达右下角需要的最小hp值
dungeon[i][j]表示这个位置的格子的数值
dp[i][j] = max(1, min(dp[i + 1][j], dp[i][j+1]) - dungeon[i][j])
#include <vector>
#include <cmath>
using namespace std;
class Solution {
public:
int calculateMinimumHP(vector<vector<int> > &dungeon) {
if(dungeon.empty())
return 0;
vector<vector<int> > dp(dungeon.size());
for(int i = 0; i < dp.size(); i++) {
dp[i].reserve(dungeon[i].size());
}
int m = dungeon.size();
int n = dungeon[0].size();
if(dungeon[m - 1][n - 1] >= 0) {
dp[m - 1][n - 1] = 1;
} else {
dp[m - 1][n - 1] = 1 - dungeon[m - 1][n - 1];
}
for(int i = m - 2; i >= 0; i--) {
dp[i][n - 1] = max(dp[i + 1][n - 1] - dungeon[i][n - 1], 1);
}
for(int j = n - 2; j >= 0; j--) {
dp[m - 1][j] = max(dp[m - 1][j + 1] - dungeon[m - 1][j], 1);
}
for(int i = m - 2; i >= 0; i--) {
for(int j = n - 2; j >= 0; j--) {
dp[i][j] = max(min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j], 1);
}
}
return dp[0][0];
}
};