-
Notifications
You must be signed in to change notification settings - Fork 0
/
174_DungeonGame.java
67 lines (56 loc) · 2.77 KB
/
174_DungeonGame.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
// The demons had captured the princess (P) and imprisoned her in the bottom-right corner of a dungeon. The dungeon consists of M x N rooms laid out in a 2D grid. Our valiant knight (K) was initially positioned in the top-left room and must fight his way through the dungeon to rescue the princess.
// The knight has an initial health point represented by a positive integer. If at any point his health point drops to 0 or below, he dies immediately.
// Some of the rooms are guarded by demons, so the knight loses health (negative integers) upon entering these rooms; other rooms are either empty (0's) or contain magic orbs that increase the knight's health (positive integers).
// In order to reach the princess as quickly as possible, the knight decides to move only rightward or downward in each step.
// Write a function to determine the knight's minimum initial health so that he is able to rescue the princess.
// For example, given the dungeon below, the initial health of the knight must be at least 7 if he follows the optimal path RIGHT-> RIGHT -> DOWN -> DOWN.
// -2 (K) -3 3
// -5 -10 1
// 10 30 -5 (P)
// Notes:
// The knight's health has no upper bound.
// Any room can contain threats or power-ups, even the first room the knight enters and the bottom-right room where the princess is imprisoned.
//O(mn) space dp
public class Solution {
public int calculateMinimumHP(int[][] dungeon) {
int row = dungeon.length;
int col = dungeon[0].length;
int[][] dp = new int[row][col];
for(int i=row - 1; i>=0; i--) {
for(int j=col - 1; j>=0; j--) {
if(i == row - 1 && j == col - 1) {
dp[i][j] = (1 - dungeon[i][j]) > 0 ? 1 - dungeon[i][j] : 1;
} else if(i == row - 1) {
dp[i][j] = dp[i][j + 1] - dungeon[i][j];
if(dp[i][j] <= 0) dp[i][j] = 1;
} else if(j == col - 1) {
dp[i][j] = Math.max(dp[i + 1][j] - dungeon[i][j], 1);
} else {
dp[i][j] = Math.max(Math.min(dp[i + 1][j], dp[i][j + 1]) - dungeon[i][j], 1);
}
}
}
return dp[0][0];
}
}
//space efficient
public class Solution {
public int calculateMinimumHP(int[][] dungeon) {
int row = dungeon.length;
int col = dungeon[0].length;
int[] dp = new int[col + 1];
for(int i=0; i< dp.length; i++) {
dp[i] = Integer.MAX_VALUE;
}
dp[col - 1] = 1;
for(int i=row - 1; i>=0 ;i--) {
for(int j=col - 1; j>=0; j--) {
dp[j] = getMin(Math.min(dp[j], dp[j + 1]) - dungeon[i][j]);
}
}
return dp[0];
}
private int getMin(int num) {
return num <= 0 ? 1 : num;
}
}