LeetCode198. 打家劫舍——动态规划

题目概述:

LeetCode198. 打家劫舍——动态规划_第1张图片

题解

class Solution {
public:
    int rob(vector<int>& nums) 
    {
        int size = nums.size();
        if (size == 1)
        {
            return nums[0];
        }
        if (size == 2)
        {
            return max(nums[0], nums[1]);
        }
        /*
        定义dp[i]考虑为下标为0~i的房子 能获得的最大金额
        显然dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        当i >= 2时
        对于房子i 有两个选择
        1.打劫当前房子,则不能打劫前一个房子 
        只能获得当前房子的nums[i]和只考虑到
        下标为0~i-2的房子能获得的最大金额 
        dp[i] = nums[i] + dp[i - 2]
        2.不打劫当前房子,获得考虑0~i-2的房子能获得的最大金额
        dp[i] = dp[i - 1]
        综上 dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
        可以利用滚动数组优化空间复杂度至O(1)
        */
        int a = nums[2], b = max(nums[0], nums[1]), c = nums[0];
        /*a dp[i] b dp[i - 1] c dp[i - 2]*/
        for (int i = 2; i < size; ++i)
        {
            a = max(b, c + nums[i]);
            c = b;
            b = a;
        }
        return a;
    }
};

你可能感兴趣的:(LeetCode刷题,动态规划,leetcode,算法)