一道题解释树形dp

最新推荐文章于 2024-11-01 15:19:21 发布

此后无良辰

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量136

点赞数 1

分类专栏：动态规划 Leetcode 文章标签： leetcode dfs 算法

本文链接：https://blog.csdn.net/qq_40493829/article/details/107820096

版权

Leetcode 同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

动态规划

3 篇文章 0 订阅

订阅专栏

题面描述

在这里插入图片描述

基本思路

树形dp
选取和不选取
分别用f和g存储
f表示选取o结点最大得到的答案
g表示不选取o结点最大得到的答案

/**
 * 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> f, g;

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

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