forked from fishercoder1534/Leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path_213.java
41 lines (38 loc) · 1.29 KB
/
_213.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
package com.fishercoder.solutions;
public class _213 {
public static class Solution1 {
/**
* Another dp problem:
* separate them into two cases:
* 1. rob from house 1 to n-1, get max1
* 2. rob from house 2 to n, get max2 take the max from the above two max
*/
public int rob(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
if (nums.length == 1) {
return nums[0];
}
if (nums.length == 2) {
return Math.max(nums[0], nums[1]);
}
int[] dp = new int[nums.length - 1];
//rob 1 to n-1
dp[0] = nums[0];
dp[1] = Math.max(nums[0], nums[1]);
for (int i = 2; i < nums.length - 1; i++) {
dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
}
int prevMax = dp[nums.length - 2];
//rob 2 to n
dp = new int[nums.length - 1];
dp[0] = nums[1];
dp[1] = Math.max(nums[1], nums[2]);
for (int i = 3; i < nums.length; i++) {
dp[i - 1] = Math.max(dp[i - 3] + nums[i], dp[i - 2]);
}
return Math.max(prevMax, dp[nums.length - 2]);
}
}
}