力扣刷题记录110.1-----337. 打家劫舍 III

最新推荐文章于 2024-06-05 21:38:06 发布

@白圭

最新推荐文章于 2024-06-05 21:38:06 发布

阅读量60

点赞数

分类专栏： # 力扣习题记录册__C++__代码随想录文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/taiyuezyh/article/details/127908818

版权

力扣习题记录册__C++__代码随想录专栏收录该内容

174 篇文章 0 订阅

订阅专栏

一、题目

在这里插入图片描述

二、代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution 
{
public:
    //对任意节点  偷和不偷 取大
    //偷和不偷  dp 有dp[0] 和dp[1]
    int rob(TreeNode* root) 
    {
        vector<int> result=robTree(root);
        return max(result[0],result[1]);

    }
    //递归 后序遍历 因为处理时需要左子树和右子树的值
    vector<int> robTree(TreeNode* process_node)
    {
        if(process_node==nullptr) return vector<int>{0,0};
        else
        {
           vector<int> left=robTree(process_node->left);
           vector<int> right=robTree(process_node->right);
           
           //分为两种情况  对于当前节点的偷和不偷
           int rob_money=0;
           int N_rob_money=0;

           //当前节点偷
           rob_money=process_node->val+left[0]+right[0];

           N_rob_money=max(left[0],left[1])+max(right[0],right[1]);

           return  vector<int>{N_rob_money,rob_money};
        }

    }
};