代码随想录算法训练营第三十四天| 198.打家劫舍， 213.打家劫舍II，337.打家劫舍III

今天是动态规划算法学习的第七天，也是打家劫舍的一天。

198.打家劫舍

题目链接：198. 打家劫舍 - 力扣（LeetCode）

这个题目是给出一个数组，小偷不能偷相邻的房间否则会报警，求小偷所能获得的最大金额。对于dp数组的含义，肯定就是小偷的最大金额；递推公式的话，小偷的金额更新有两种方式，一种是不偷dp[i]一种是偷dp[i]，选择其中的较大值，所以dp[i]=max(dp[i-1],dp[i-2]+nums[i]);初始化方式dp[0]=nums[0],dp[1]=max(nums[0],nums[1])。手动递推一下也是没问题的。具体代码如下所示：
 

class Solution {
public:
    int rob(vector<int>& nums) {
        if(nums.size()==1) return nums[0];
        vector<int> dp(nums.size(),0);
        dp[0]=nums[0];
        dp[1]=max(nums[0],nums[1]);
        for(int i=2;i<nums.size();i++)
        {
            dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
        }
        return dp[nums.size()-1];

    }
};

213.打家劫舍II

题目链接：213. 打家劫舍 II - 力扣（LeetCode）

这个题目在打家劫舍的基础上加上了一个限定条件，即这个地方所有的房屋都 围成一圈 ，这意味着第一个房屋和最后一个房屋是紧挨着的。我自己一开始并没有想到合理的解法，看了题解才学会的。直接处理成两种情况，分别是去首和去尾，最后输出两种情况的较大值即可。

具体代码实现如下所示：

class Solution {
public:
int robamount(vector<int> &vec)
{
    if(vec.size()==1) return vec[0];
    vector<int> dp(vec.size(),0);
    dp[0]=vec[0];
    dp[1]=max(vec[0],vec[1]);
    for(int i=2;i<vec.size();i++)
    {
        dp[i]=max(dp[i-1],dp[i-2]+vec[i]);
    }
    return dp[vec.size()-1];
}
    int rob(vector<int>& nums) {
         if(nums.size()==1) return nums[0];
        vector<int> vec1(nums.begin(),nums.end()-1);
        vector<int> vec2(nums.begin()+1,nums.end());
        int result1=robamount(vec1);
        int result2=robamount(vec2);
        return max(result1,result2);
    }
};

337.打家劫舍 III 

题目链接：337. 打家劫舍 III - 力扣（LeetCode）

这个题目就更难了，是一个树形dp，即不能偷相邻两节点的房间，获取最大的金额。首先对于二叉树，我们使用后序遍历。dp数组存储两个结果，dp[0]返回的是不偷当前房间的金额数，dp[1]返回的是偷当前房间的金额数。递归的终止条件是遇到空结点，返回{0,0}，然后求解左节点的dp数组和右节点的dp数组，获取偷当前节点和不偷当前节点的金额数，并以dp数组的形式返回给上一层。这是第一次接触树状dp的题目，感觉还是挺不好想的。具体代码实现如下图所示：
 

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
vector<int> order(TreeNode* root)
{
    if(root==NULL) return vector<int>{0,0};
    vector<int> left=order(root->left);
    vector<int> right=order(root->right);
    int val1=root->val+left[0]+right[0];
    int val2=max(left[0],left[1])+max(right[0],right[1]);
    return {val2,val1};
}
    int rob(TreeNode* root) {
        if(root==NULL) return 0;
        vector<int> result=order(root);
        return max(result[0],result[1]);
    }
};