明显是DP,奈何想了很久想不出二次方复杂度公式,原来是要逆向推导
dp[i][j]的状态从dp[i][j+1]与dp[i+1][j]得来,代表从当前到最后所需最小HP。
当前点加血(>0),则需要最小HP会减少;
当前点掉血(<0),则会增加HP需求。
后面的HP需求是一个子问题,与前面无关,可以递推,建立递推公式即可。
class Solution {
public:
int calculateMinimumHP(vector<vector<int> > &dungeon) {
int m=dungeon.size(),n=dungeon[0].size();
vector<int> dp[2];
dp[0].resize(n);
dp[1].resize(n);
dp[1][n-1]=max(-dungeon[m-1][n-1],0);
for (int j=n-2;j>=0;--j){
dp[1][j]=max(dp[1][j+1]-dungeon[m-1][j],0);
}
int pos=0;
for (int i=m-2;i>=0;--i){
dp[pos][n-1]=max(dp[1-pos][n-1]-dungeon[i][n-1],0);
for (int j=n-2;j>=0;--j){
dp[pos][j]=max(min(dp[pos][j+1],dp[1-pos][j])-dungeon[i][j],0);
}
pos=1-pos;
}
return dp[1-pos][0]+1;
}
};