【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]);
    }
};

你可能感兴趣的:(LeetCode)