Leetcode 337. 打家劫舍 III 树形dp

最新推荐文章于 2024-09-20 23:05:22 发布

wwxy261

最新推荐文章于 2024-09-20 23:05:22 发布

阅读量141

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/107813882

版权

算法专栏收录该内容

3633 篇文章 120 订阅

订阅专栏

用f(root)表示从root节点出发能盗取的最高金额

f(root) = max(f(root->left)+f(root->right), root->val + f(root->left->left)+f(root->left->right) + f(root->right->left) + f(root->right->right))

如果直接这样搜索，为有大量的重复搜索，所以要用hashmap记录状态

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<TreeNode*,int> hashmap;
    int rob(TreeNode* root) {
        // rob(root) 表示从根节点出发的最高金额
        if(root==NULL) return 0;
        if(hashmap.count(root)) return hashmap[root];
        int money = root->val;
        if(root->left){
            money += rob(root->left->left)+rob(root->left->right);
        }
        if(root->right){
            money += rob(root->right->right)+rob(root->right->left);
        }
        return max(money,rob(root->left)+rob(root->right));
    }
};

记忆话优化

class Solution {
public:
    unordered_map<TreeNode*,int> hashmap;
    int rob(TreeNode* root) {
        // rob(root) 表示从根节点出发的最高金额
        if(root==NULL) return 0;
        if(hashmap.count(root)) return hashmap[root];
        int money = root->val;
        if(root->left){
            money += rob(root->left->left)+rob(root->left->right);
        }
        if(root->right){
            money += rob(root->right->right)+rob(root->right->left);
        }
        int val = max(money,rob(root->left)+rob(root->right));
        hashmap[root] = val;
        return val;
    }
};