LeetCode算法题解 198-打家劫舍

题目描述

题解:

动态规划方程:
dp[i] = max(dp[i-2] + a[i],dp[i-1])
对于第i号房屋,有两种选择:1. 拿这家的a[i],然后考虑dp[i-2]的(不能拿相邻的)2. 不拿这家的,考虑dp[i-1],从中选择大的。

代码:

class Solution {
public:
    int rob(vector& nums) {
        int len = nums.size();
        if(len == 0)
        {
            return 0;
        }
        if(len == 1)
        {
            return nums[0];
        }
        vector  dp(len + 1);
        dp[0] = nums[0];  dp[1] = max(nums[0],nums[1]);
        for(int i = 2; i < len; i++)
        {
            dp[i] = max(dp[i-2]+nums[i],dp[i-1]);
        } 
        return dp[len-1];
    }
};

你可能感兴趣的:(#,LeetCode)