2023-09-18 LeetCode每日一题(打家劫舍 III)

2023-09-18每日一题

一、题目编号

337. 打家劫舍 III

二、题目链接

点击跳转到题目位置

三、题目描述

小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为 root 。

除了 root 之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ,房屋将自动报警。

给定二叉树的 root 。返回 在不触动警报的情况下 ,小偷能够盗取的最高金额 。

示例 1:
2023-09-18 LeetCode每日一题(打家劫舍 III)_第1张图片

示例 2:
2023-09-18 LeetCode每日一题(打家劫舍 III)_第2张图片
提示:

  • 树的节点数在 [1, 104] 范围内
  • 0 <= Node.val <= 104

四、解题代码

class Solution {
public:
    unordered_map <TreeNode*, int> f, g;

    void dfs(TreeNode* node) {
        if (!node) {
            return;
        }
        dfs(node->left);
        dfs(node->right);
        f[node] = node->val + g[node->left] + g[node->right];
        g[node] = max(f[node->left], g[node->left]) + max(f[node->right], g[node->right]);
    }

    int rob(TreeNode* root) {
        dfs(root);
        return max(f[root], g[root]);
    }
};


五、解题思路

(1) 树型dp题目

你可能感兴趣的:(LeetCode每日一题,leetcode,数据结构,算法)