代码随想录训练营第39天 || 198. 打家劫舍 213. 打家劫舍 II 337. 打家劫舍 III-优快云博客

本文链接：https://blog.youkuaiyun.com/dkingf/article/details/147440017

198. 打家劫舍

思路：

动规五部曲：

1.dp数组及其下标的意义：dp数组表示当前房屋下偷与不偷的最大盗取金额

2.确定递推公式：因为盗取房屋只能间隔盗取，并且还要取最大值。所以每个房屋都有盗取和不盗取两个选择，盗不盗取取决于金额，所以递推公式为dp[ i ] = max（dp[ i-1] (不盗取），dp[i-2 ]+nums[ i ]（盗取））

3.dp数组的初始化：因为递推公式要取最大盗取金额，同时金额都为非负数，所以初始化为最小值0

4.遍历顺序：从前往后

代码：

class Solution {
public:
    int rob(vector<int>& nums) {
        vector<int> dp(nums.size(),0);
        if(nums.size() == 0)return 0;
        if(nums.size()==1)return nums[0];//一定不要忘了无法用递推公式的情况
        //dp数组的初始化
        dp[0] = nums[0];
        dp[1] = max(nums[0],nums[1]);
        for(int i =2;i <nums.size();i++)
        {
            dp[i] = max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[nums.size()-1];
    }
};

遇到的问题：

对于一开始如果nums的个数小于三，那么就无法应用递推公式，需要手动判断

213. 打家劫舍 II

思路：

此题与上一题不同的是，这次的家是环形的，而非线性。因为相邻就不偷，所以第一个和最后一个只有一个可以偷，所以分两种情况

1.dp数组及其下标的意义：

dp数组表示当前序号房屋下偷与不偷的最大盗取金额

2.确定递推公式：

dp[ i ] = max（dp[ i-1] (不盗取），dp[i-2 ]+nums[ i ]（盗取））

3.dp数组的初始化：因为递推公式要取最大盗取金额，同时金额都为非负数，所以初始化为最小值0

4.遍历顺序：从前往后

代码：

class Solution {
public:
    int rob(vector<int>& nums) {
        if(nums.size() == 0)return 0;
        if(nums.size()==1)return nums[0];
        int result1 = caozuo(nums,1,nums.size()-1);
        int result2 = caozuo(nums,0,nums.size()-2);
        return max(result1,result2);
    }
    int caozuo(vector<int>& nums ,int start,int end)
    {
        if(start == end)return nums[start];

        vector<int> dp(nums.size(),0);//此处相当于复制了一个dp数组，因为start和end是原来的dp数组截取的下标，所以长度不能动
        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];//此处因为最后的值为end
    }
};

遇到的问题：

对于在分情况讨论只有头房子和尾房子时，处理dp数组的方式与不分情况是一样的

337. 打家劫舍 III

思路：

1.dp数组的意义：dp数组下标为0记录不偷该节点所得到的的最大金钱，下标为1记录偷该节点所得到的的最大金钱。

2.中止条件：cur ==NULL遇到叶子节点

3.遍历顺序：因为是二叉树，所以遍历顺序从前中后遍历中选择，选择后序遍历，因为当前节点偷不偷取决于子节点

代码：

class Solution{
public:
    int rob(TreeNode* root) {
    //dp数组采用一个动态数组,内部只有两个元素，0和1，
    vector<int> result = caozuo(root);
    return max(result[0],result[1]);

}
    vector<int> caozuo(TreeNode* cur)
    {
        if(cur == NULL)return {0,0};
        //后序遍历
        //左
        vector<int>left = caozuo(cur->left);
        //右
        vector<int>right = caozuo(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};//{ }是列表的初始化来赋值数组
    }
};