【算法/动态规划/打家劫舍问题】题解+详细备注（共3题）

198.打家劫舍

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if(n == 1) return nums[0];
        if(n == 2) return max(nums[0],nums[1]);
		// 1.确定dp数组以及下标的含义
		// dp[i]表示0-i的房屋，一夜之内可以偷到的最大金额为dp[i]
        vector<int> dp(n);
		
		//3.dp数组初始化
        dp[0] = nums[0];
        dp[1] = max(nums[0],nums[1]);
		
		// 4.确定遍历顺序：按照房屋遍历
        for(int i = 2;i<n;++i){
        	// 2.确定递推公式：dp[i] = max(dp[i-1],dp[i-2] + nums[i]);
            dp[i] = max(dp[i-1],dp[i-2] + nums[i]);
        }
        return dp[n-1];
    }
};

213.打家劫舍II

class Solution {
public:
	// 子方法与198.打家劫舍一致
    int rob1(vector<int>& nums) {
        int n = nums.size();
        if(n == 1) return nums[0];
        if(n == 2) return max(nums[0],nums[1]);

        vector<int> dp(n);
        dp[0] = nums[0];
        dp[1] = max(nums[0],nums[1]);

        for(int i = 2;i<n;++i){
            dp[i] = max(dp[i-1],dp[i-2] + nums[i]);
        }
        return dp[n-1];
    }
    // 198.打家劫舍进阶版，拆分成两次计算即可
    int rob(vector<int>& nums) {
        int n = nums.size();
        if(1 == n) return nums[0];
        if(2 == n) return max(nums[0],nums[1]);
        vector<int> nums1(nums.begin(),nums.end()-1);
        vector<int> nums2(nums.begin()+1,nums.end());

        return max(rob1(nums1),rob1(nums2));
    }
};

337.打家劫舍III

class Solution {
public:
	// 树要第一时间想到遍历方式：本题使用后续遍历
    int rob(TreeNode* root) {
        vector<int> result = robTree(root);
        return max(result[0], result[1]);
    }
    // 递归返回值，0：不偷，1：偷
    vector<int> robTree(TreeNode* cur) {
        if (cur == NULL) return vector<int>{0, 0};
        vector<int> left = robTree(cur->left);
        vector<int> right = robTree(cur->right);
        // 偷当前节点
        int val1 = cur->val + left[0] + right[0];
        // 不偷当前节点
        int val2 = max(left[0], left[1]) + max(right[0], right[1]);
        return {val2, val1};
    }
};