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<int>& nums) {
        int len = nums.size();
        if(len == 0)
        {
            return 0;
        }
        if(len == 1)
        {
            return nums[0];
        }
        vector <int> 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];
    }
};