1、问题分析

题目链接：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;
}

运行结果
在这里插入图片描述

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

3、总结

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