动态规划,代码如下:
int calculateMinimumHP(vector<vector<int> > &dungeon) {
const int m = dungeon.size();
const int n = dungeon[0].size();
int dp[m][n];
dp[m - 1][n - 1] = max(1, 1 - dungeon[m - 1][n - 1]);
for(int i = n - 2; i >= 0; --i)
dp[m - 1][i] = max(1, dp[m - 1][i + 1] - dungeon[m - 1][i]);
for(int i = m - 2; i >= 0; --i)
dp[i][n - 1] = max(1, dp[i + 1][n - 1] - dungeon[i][n - 1]);
for(int i = m - 2; i >= 0; --i)
for(int j = n - 2; j >= 0; --j)
dp[i][j] = max(1, min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j]);
return dp[0][0];
}
回滚数组优化下,代码如下:
int calculateMinimumHP(vector<vector<int> > &dungeon) {
vector<int> dp(dungeon[0].size() + 1, INT_MAX);
dp[dungeon[0].size() - 1] = 1;
for(int i = dungeon.size() - 1; i >= 0; --i)
for(int j = dungeon[0].size() - 1; j >= 0; --j)
dp[j] = max(1, min(dp[j], dp[j + 1]) - dungeon[i][j]);
return dp[0];
}
看了下discuss,空间还可以优化到O(1),trick就是利用dungeon数组,代码基本和第一个一样。