213. House Robber II
- 2016/Oct/19,做了不止一次,本次独立解出,大概耗时13分钟
本题并不难,跟198. House Robber解法基本一样,都是用dynamic programming解,需要做一点小改动:
- dp两次,第一次不算首元素,第二次不算尾元素,求这两次的最大值
代码如下:
class Solution {
public:
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> dp1(nums.size() - 1, 0);
vector<int> dp2(nums.size() - 1, 0);
dp1[0] = nums[0];
dp1[1] = max(nums[0], nums[1]);
for (int i = 2; i < nums.size() - 1; ++i) {
dp1[i] = max(dp1[i-1], dp1[i-2] + nums[i]);
}
dp2[0] = nums[1];
dp2[1] = max(nums[1], nums[2]);
for (int i = 3; i < nums.size(); ++i) {
dp2[i - 1] = max(dp2[i-2], dp2[i-3] + nums[i]);
}
return max(dp1.back(), dp2.back());
}
};