题目地址:
https://leetcode.com/problems/dungeon-game/
给定一个二维矩阵 A A A,一个骑士从左上角出发,只能向右或者向下走一格,要求走到右下角。每次走到一格的时候,他的血量就会改变当前格子里的数字那么多,数字是正的血量就增加,否则就减少。要求路径上不能血量变为 0 0 0或负。问他出发时候的血量至少要是多少。左上角出发时候的格子也会影响血量。
思路是动态规划。设 f [ i ] [ j ] f[i][j] f[i][j]是到达 A [ i ] [ j ] A[i][j] A[i][j]之前必须至少要有的血量。那么到达 A [ x ] [ y ] A[x][y] A[x][y]所需要的血量,就要看 f [ x + 1 ] [ y ] f[x+1][y] f[x+1][y]和 f [ x ] [ y + 1 ] f[x][y+1] f[x][y+1]的最小者,哪个小就走哪条路,所以,走到这个位置之前的血量必须不但能支持从这个位置继续向下走的血量,也需要支持 A [ x ] [ y ] A[x][y] A[x][y]的消耗,所以到这个位置之前的血量 z z z应该满足 z + A [ x ] [ y ] ≥ min { f [ x + 1 ] [ y ] , f [ x ] [ y + 1 ] } z+A[x][y]\ge \min\{f[x+1][y],f[x][y+1]\} z+A[x][y]≥min{f[x+1][y],f[x][y+1]},当然 z > 0 z>0 z>0也要成立,所以就有: f [ x ] [ y ] = max { 1 , min { f [ x + 1 ] [ y ] , f [ x ] [ y + 1 ] } − A [ x ] [ y ] } f[x][y]=\max\{1,\min\{f[x+1][y],f[x][y+1]\}-A[x][y]\} f[x][y]=max{1,min{f[x+1][y],f[x][y+1]}−A[x][y]}最后返回 f [ 0 ] [ 0 ] f[0][0] f[0][0]即可。代码如下:
public class Solution {
public int calculateMinimumHP(int[][] dungeon) {
int m = dungeon.length, n = dungeon[0].length;
int[][] dp = new int[m][n];
for (int i = m - 1; i >= 0; i--) {
for (int j = n - 1; j >= 0; j--) {
if (i == m - 1 && j == n - 1) {
dp[i][j] = Math.max(1, 1 - dungeon[i][j]);
} else {
int minBlood = Integer.MAX_VALUE;
if (i < m - 1) {
minBlood = Math.min(minBlood, dp[i + 1][j]);
}
if (j < n - 1) {
minBlood = Math.min(minBlood, dp[i][j + 1]);
}
dp[i][j] = Math.max(1, minBlood - dungeon[i][j]);
}
}
}
return dp[0][0];
}
}
时空复杂度 O ( m n ) O(mn) O(mn)。