算法训练第四十八天

总结：今日刷了打家劫舍三道题

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

总结：第三道自己不太会，要利用二叉树的后序遍历来解决问题，从底向上，并且要将dp数组设置为两种状态，偷与不偷。

代码：

class Solution {
public:
    vector robTree(TreeNode* root)
    {
        if(root == nullptr)
        return vector{0,0};

        vector leftdp = robTree(root->left);
        vector rightdp = robTree(root->right);

        int val1 = root->val + leftdp[0] + rightdp[0];
        int val2 = max(leftdp[0],leftdp[1]) + max(rightdp[0],rightdp[1]);

        return vector{val2,val1};
    }
    int rob(TreeNode* root) {
        vector res = robTree(root);
        return max(res[0],res[1]);
    }
};

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

代码：

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

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

总结：两种情况进行讨论，第一家偷或者不偷这两种情况

class Solution {
public:
    int rob(vector& nums) {
        if(nums.size() == 1)
        return nums[0];
        vector dp1(nums.size() + 1,0);
        vector dp2(nums.size() + 1,0);

        dp1[1] = nums[0];
        dp1[2] = max(nums[0],nums[1]);

        dp2[1] = 0;
        dp2[2] = nums[1];

        for(int i = 3;i <= nums.size() - 1;i++)
        {
            dp1[i] = max(dp1[i - 1],dp1[i - 2] + nums[i - 1]);
        }
        for(int i = 3;i <= nums.size();i++)
        {
            dp2[i] = max(dp2[i - 1],dp2[i - 2] + nums[i - 1]);
        }

        return max(dp1[nums.size() - 1],dp2[nums.size()]);
    }
};