1. 程式人生 > >【LeetCode】213. 打家劫舍 II——dp

【LeetCode】213. 打家劫舍 II——dp

題目

解答

和之前的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]); } };