代码随想录算法训练营刷题笔记

213. 打家劫舍 II

这个是自己写的

class Solution {
public:
    int rob(vector<int>& nums) {
        if (nums.size() == 0) return 0;
        if (nums.size() == 1) return nums[0];
        // dp[i]：考虑下标i（包括i）以内的房屋，最多可以偷窃的金额为dp[i]
        // dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
        vector<int> dp(nums.size());
        // skip index 0 first
        int res = 0;
        dp[0] = 0;
        dp[1] = nums[1];
        for (int i = 2; i < nums.size(); i++) {
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
        }
        res = dp[nums.size()-1];
        dp[0] = nums[0];
        dp[1] = max(nums[0], nums[1]);
        
        for (int i = 2; i < nums.size() - 1; i++) {
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
        }

        res = max(res,dp[nums.size()-2]);
        return res;
    }
};