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 | class Solution {
public:
static constexpr int SIZE = 101;
int dp(vector<int>& nums) {
int x[SIZE];
if (nums.size() == 1)
return nums[0];
x[0] = nums[0];
x[1] = max(nums[0], nums[1]);
for (int i = 2; i < nums.size(); i++)
x[i] = max(x[i - 1], x[i - 2] + nums[i]);
return x[nums.size() - 1];
}
int rob(vector<int>& nums) {
if (nums.size() == 0)
return 0;
if (nums.size() == 1)
return nums[0];
if (nums.size() == 2)
return max(nums[0], nums[1]);
vector<int> nums1(nums.begin(), nums.end() - 1);
vector<int> nums2(nums.begin() + 1, nums.end());
int dp1 = dp(nums1);
int dp2 = dp(nums2);
return max(dp1, dp2);
}
};
|