【c++刷题笔记-动态规划】day40： 198.打家劫舍、 213.打家劫舍II 、 337.打家劫舍III-CSDN博客

本文链接：https://blog.csdn.net/qq_48662659/article/details/140434650

198. 打家劫舍 - 力扣（LeetCode）

思路：两种状态，在偷和不偷之间判断最大的值，然后递推

重点： dp[i]=max(dp[i-1],dp[i-2]+nums[i]);

class Solution {
public:
    int rob(vector<int>& nums) {
        int n=nums.size();
        vector<int>dp(n);
        if(n==1){
            return nums[0];
        }
        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];
    }
};

213. 打家劫舍 II - 力扣（LeetCode）

思路：分为两种情况讨论，[0,n-2],[1,n-1]，取最大值

重点：dp[i]=max(dp[i-1],dp[i-2]+nums[i]);

class Solution {
public:
    int rob(vector<int>& nums) {
        int n=nums.size();
        if(n==0){
            return 0;
        }
        if(n==1){
            return nums[0];
        }
        int a=fun(nums,0,n-2);//第一种情况
        int b=fun(nums,1,n-1);//第二种情况
        return max(a,b);
    }
    int fun(vector<int>& nums,int start,int end){
        if(start==end){
            return nums[start];
        }
        vector<int>dp(nums.size());
        dp[start]=nums[start];
        dp[start+1]=max(nums[start],nums[start+1]);
        for(int i=start+2;i<=end;i++){
            dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[end];
    }
};

337. 打家劫舍 III - 力扣（LeetCode）

思路：长度为2的数组，下标0表示不偷，下标1表示偷。利用这两个状态进行推导

class Solution {
public:
    int rob(TreeNode* root) {
        vector<int>ans=fun(root);
        return max(ans[0],ans[1]);
    }
    //长度为2的数组，下标0表示不偷，下标1表示偷
    vector<int>fun(TreeNode* root){
        if(root==nullptr){
            return {0,0};
        }
        vector<int>left=fun(root->left);
        vector<int>right=fun(root->right);
        //偷cur，那么就不能偷左右节点。
        int val1=root->val+left[0]+right[0];
        //不偷cur，那么可以偷也可以不偷左右节点，则取较大的情况
        int val2=max(left[0],left[1])+max(right[0],right[1]);
        return {val2,val1};
    }
};