337.打家劫舍 III

咔咔咔的

已于 2024-04-21 09:49:30 修改

阅读量187

点赞数 2

分类专栏： leetcode 文章标签： c++

于 2024-03-31 05:30:00 首次发布

本文链接：https://blog.csdn.net/weixin_45256307/article/details/137006305

版权

leetcode 专栏收录该内容

597 篇文章 2 订阅

订阅专栏

文章介绍了在给定的二叉树结构中，使用动态规划和记忆化方法解决打家劫舍问题，即在不偷父节点的情况下最大化偷窃值。作者提供了两种方法：一种是递归记忆化，另一种是通过构建长度为2的数组表示偷窃决策的结果。

摘要由CSDN通过智能技术生成

337.打家劫舍 III

题目链接：337.打家劫舍 III

代码如下：

/**
 * 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) {}
 * };
 */
//参考：https://www.programmercarl.com/0337.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8DIII.html#%E6%80%9D%E8%B7%AF
class Solution {
public:
    //记忆化递推
    // int rob(TreeNode* root) 
    // {
    //     if(root==nullptr)   return 0;
    //     //当前节点为叶子节点
    //     if(root->left==nullptr&&root->right==nullptr)   return root->val;
    //     if(um[root]) return um[root];//如果um上有记录就返回即可
    //     //偷父节点
    //     int val1=root->val;
    //     if(root->left)
    //         val1+=rob(root->left->left)+rob(root->left->right);
    //     if(root->right)
    //         val1+=rob(root->right->left)+rob(root->right->right);
    //     //不偷父节点
    //     int val2=rob(root->left)+rob(root->right);
    //     um[root]=max(val1,val2);//对当前节点所求的和进行记忆
    //     return max(val1,val2);
    // }

    //动态规划
    int rob(TreeNode* root) 
    {
        vector<int> res=robTree(root);
        return max(res[0],res[1]);
    }

    //长度为2的数组，0：不偷，1：偷
    vector<int> robTree(TreeNode* cur)
    {
        if(cur==nullptr)    return {0,0};
        vector<int> left=robTree(cur->left);
        vector<int> right=robTree(cur->right);
        //偷cur，那么不能偷左右节点
        int val1=cur->val+left[0]+right[0];
        //不偷cur，那么可以偷也可以不偷左右节点，取较大的情况
        int val2=max(left[0],left[1])+max(right[0],right[1]);
        return {val2,val1};
    }

private:
    unordered_map<TreeNode*,int> um;
};