-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathHouse Robber II.java
38 lines (27 loc) · 1.05 KB
/
House Robber II.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
// https://leetcode.com/problems/house-robber-ii/
class Solution {
public int rob(int[] nums) {
if(nums.length == 1) return nums[0];
if(nums.length == 2) return Math.max(nums[0], nums[1]);
if(nums.length == 3) return Math.max(nums[0], Math.max(nums[1], nums[2]));
int dp[] = new int[nums.length];
dp[0] = nums[0];
dp[1] = nums[1];
dp[2] = nums[0] + nums[2];
for(int i = 3; i < nums.length - 1; i++)
{
dp[i] += Math.max(dp[i - 2], dp[i - 3]) + nums[i];
}
int ans1 = Math.max(dp[dp.length - 2], dp[dp.length - 3]);
dp[1] = nums[1];
dp[2] = nums[2];
dp[3] = nums[1] + nums[3];
for(int i = 4; i < nums.length; i++)
{
dp[i] = 0;
dp[i] += Math.max(dp[i - 2], dp[i - 3]) + nums[i];
}
int ans2 = Math.max(dp[dp.length - 1], dp[dp.length - 2]);
return Math.max(ans1, ans2);
}
}