Leetcode 337. 打家劫舍 III

分手后的思念是犯贱 2023-02-14 03:07 34阅读 0赞

### Leetcode 337. 打家劫舍 III ###

*  1、问题分析
 *  2、问题解决
 *  3、总结

# 1、问题分析 #

题目链接：[https://leetcode-cn.com/problems/house-robber-iii/][https_leetcode-cn.com_problems_house-robber-iii]  
  本质上就是一个二叉树的遍历问题。代码我已经进行了详细的注释，理解应该没有问题，读者可以作为参考，如果看不懂(可以多看几遍)，欢迎留言哦！我看到会解答一下。

# 2、问题解决 #

笔者以`C++`方式解决。

#include "iostream"
    
    using namespace std;
    
    #include "algorithm"
    #include "vector"
    #include "queue"
    #include "set"
    #include "map"
    #include "string"
    #include "stack"
    
    /** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */
    
    struct TreeNode { 
        int val;
        TreeNode *left;
        TreeNode *right;
    
        TreeNode(int x) : val(x), left(NULL), right(NULL) { }
    };
    
    class Solution { 
    private:
        // 保存访问过的节点能够盗取的最高金额
        // key 为：节点指针地址 和 父节点是否访问 的结合体
        map<pair<TreeNode *,int>, int> dp_map;
    public:
        int rob(TreeNode *root) { 
            // 如果树为空，则直接返回 0
            if (root == NULL) { 
                return 0;
            }
    
            // 不抢劫根节点时再去抢劫左右子树
            int dontVisit = dealChen(root->left, 0) + dealChen(root->right, 0);
            // 抢劫根节点，再去抢劫左右子树
            int visited = dealChen(root->left, 1) + dealChen(root->right, 1) + root->val;
            // 选择其中较大的
            return max(dontVisit, visited);
    
        }
    
        /** * 根据父节点的状态抢劫树 * @param root * @param flag 0 代表父节点没有被抢劫; 1 代表父节点被抢劫 * @return */
        int dealChen(TreeNode *root, int flag) { 
            // 如果节点为空，则直接返回 0
            if (root == NULL) { 
                return 0;
            }
    
            // 首先查找该节点是否计算过，
            auto iterator1 = dp_map.find(make_pair(root,flag));
            // 如果计算过，则直接返回计算过的值
            if (iterator1 != dp_map.end()) { 
                return iterator1->second;
            }
    
            // 不抢劫本节点，直接抢劫左右子树
            int left = dealChen(root->left, 0);
            int right = dealChen(root->right, 0);
            // 不抢劫本节点获取的最高金额
            int dontVisit = left + right;
    
            // 如果父节点已经被抢劫过，则直接返回 不抢劫本节点获取的最高金额
            if (flag == 1) { 
                // 同时保存该值
                dp_map.insert(make_pair(make_pair(root, flag), dontVisit));
                return dontVisit;
            }
    
            // 抢劫该节点之后，再去抢劫左右子树
            int leftVisited = dealChen(root->left, 1);
            int rightVisited = dealChen(root->right, 1);
            // 抢劫本节点 和 不抢劫本节点 选择其中较大的值
            int visited = max(dontVisit, root->val + leftVisited + rightVisited);
            // 同时保存该值
            dp_map.insert(make_pair(make_pair(root, flag), visited));
            // 返回计算结果
            return visited;
        }
    };
    
    int main() { 
    
    // TreeNode *pNode1 = new TreeNode(3);
    // TreeNode *pNode2 = new TreeNode(2);
    // TreeNode *pNode3 = new TreeNode(3);
    // TreeNode *pNode4 = new TreeNode(3);
    // TreeNode *pNode5 = new TreeNode(1);
    //
    // pNode1->left = pNode2;
    // pNode1->right = pNode3;
    // pNode2->right = pNode4;
    // pNode3->right = pNode5;
    
        TreeNode *pNode11 = new TreeNode(3);
        TreeNode *pNode12 = new TreeNode(4);
        TreeNode *pNode13 = new TreeNode(5);
        TreeNode *pNode14 = new TreeNode(1);
        TreeNode *pNode15 = new TreeNode(3);
        TreeNode *pNode16 = new TreeNode(1);
    
        pNode11->left = pNode12;
        pNode11->right = pNode13;
        pNode12->left = pNode14;
        pNode12->right = pNode15;
        pNode13->right = pNode16;
    
    
    
        Solution *pSolution = new Solution;
        int i = pSolution->rob(pNode11);
        cout << i << endl;
        system("pause");
        return 0;
    }

运行结果  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3_size_16_color_FFFFFF_t_70]

有点菜，有时间再优化一下。

# 3、总结 #

难得有时间刷一波`LeetCode`, 这次做一个系统的记录，等以后复习的时候可以有章可循，同时也期待各位读者给出的建议。算法真的是一个照妖镜，原来感觉自己也还行吧，但是算法分分钟教你做人。前人栽树，后人乘凉。在学习算法的过程中，看了前辈的成果，受益匪浅。  
感谢各位前辈的辛勤付出，让我们少走了很多的弯路！  
哪怕只有一个人从我的博客受益，我也知足了。  
点个赞再走呗！欢迎留言哦！

[https_leetcode-cn.com_problems_house-robber-iii]: https://leetcode-cn.com/problems/house-robber-iii/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200531220655286.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3,size_16,color_FFFFFF,t_70