forked from sachuverma/DataStructures-Algorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path14. House Robber II.cpp
66 lines (50 loc) · 1.88 KB
/
14. House Robber II.cpp
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
/*
House Robber II
===============
You are a professional robber planning to rob houses along a street. Each house has a certain amount of money stashed. All houses at this place are arranged in a circle. That means the first house is the neighbor of the last one. Meanwhile, adjacent houses have a security system connected, and it will automatically contact the police if two adjacent houses were broken into on the same night.
Given a list of non-negative integers nums representing the amount of money of each house, return the maximum amount of money you can rob tonight without alerting the police.
Example 1:
Input: nums = [2,3,2]
Output: 3
Explanation: You cannot rob house 1 (money = 2) and then rob house 3 (money = 2), because they are adjacent houses.
Example 2:
Input: nums = [1,2,3,1]
Output: 4
Explanation: Rob house 1 (money = 1) and then rob house 3 (money = 3).
Total amount you can rob = 1 + 3 = 4.
Example 3:
Input: nums = [0]
Output: 0
Constraints:
1 <= nums.length <= 100
0 <= nums[i] <= 1000
Hint #1
Since House[1] and House[n] are adjacent, they cannot be robbed together. Therefore, the problem becomes to rob either House[1]-House[n-1] or House[2]-House[n], depending on which choice offers more money. Now the problem has degenerated to the House Robber, which is already been solved.
*/
class Solution
{
int solve(vector<int> &nums, int s, int e)
{
int n = e - s + 1;
if (n == 0)
return 0;
vector<int> dp(n, 0);
dp[0] = nums[s];
dp[1] = max(nums[s], nums[s + 1]);
for (int i = s + 2; i <= e; ++i)
dp[i - s] = max(dp[i - s - 1], dp[i - s - 2] + nums[i]);
return dp[e - s];
}
public:
int rob(vector<int> &nums)
{
int n = nums.size();
if (n == 0)
return 0;
if (n == 1)
return nums[0];
if (n == 2)
return max(nums[0], nums[1]);
return max(solve(nums, 0, n - 2), solve(nums, 1, n - 1));
}
};