【LeetCode】213. 打家劫舍 II——dp
阿新 • • 發佈:2019-02-02
題目
解答
和之前的198.打家劫舍類似,但有個限制就是陣列首尾相鄰,因此分兩種情況:從0開始和從1開始。
程式碼中dp1代表從0開始,dp2代表從1開始,均有如下狀態轉移方程:
f(n) = max{f(n-1), f(n-2)+nums(n)}
(f(n)代表到第n個房屋為止獲得的最大金額)
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]);
if (nums.size() == 3)
return max(max(nums[0], nums[1]), nums[2]);
int dp1[2] = {0};
dp1[0] = nums[0];
dp1[1] = max(nums[0], nums[1]);
for (int i = 2; i < nums.size()-1; i++) {
int temp = dp1[1];
dp1[1] = max(dp1[0]+nums[i], dp1[1]);
dp1[0] = temp;
}
int dp2[2] = {0};
dp2[0] = nums[1];
dp2[1] = max(nums[1], nums[2]);
for (int i = 3; i < nums.size(); i++) {
int temp = dp2[1];
dp2[1] = max(dp2[0 ]+nums[i], dp2[1]);
dp2[0] = temp;
}
return max(dp1[1], dp2[1]);
}
};