LeetCode第 337 题：打家劫舍III(C++)

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

如果偷了父节点，就不能动子节点。

要找最大值，遍历不可避免，第一反应是回溯法，不过因为树是无序的，并不存在剪枝规则，所以回溯其实就是不停的递归。

• 根节点被偷，子节点必然不能偷；
• 根节点没被偷，子节点可偷可不偷。

暴力递归

参考这个题解：
三种方法解决树形动态规划问题-从入门级代码到高效树形动态规划代码实现 - 打家劫舍 III - 力扣（LeetCode）
爷爷，儿子，孙子的例子非常形象。

代码还是有点抽象的，而且会超时

class Solution {
public:
    int rob(TreeNode* root) {
        if(!root) return 0;
        int money = root->val;
        if(root->left)
            money += rob(root->left->left) + rob(root->left->right);
        if(root->right)
            money += rob(root->right->left) + rob(root->right->right);
        return max(money, rob(root->left) + rob(root->right));
    }
};

简单优化

既然是重复计算，那就可以一遍计算一遍记录嘛，计算之前查表看之前有没有计算过，如果就算过直接查表获取值就可以了。

class Solution {
public:
    unordered_map<TreeNode*, int> map;
    int rob(TreeNode* root) {
        if(!root) return 0;
        if(!root->left && !root->right) return root->val;
        if(map[root]) return map[root];

        int money1 = rob(root->left) + rob(root->right);//不偷父节点
        //偷父节点，该父节点的子节点就不能偷了
        int money2 = root->val;
        if(root->left)  money2 += rob(root->left->left) + rob(root->left->right);
        if(root->right)  money2 += rob(root->right->left) + rob(root->right->right);

        map[root] = max(money1, money2);
        return map[root];
    }
};