leetcode - 337. House Robber III

最新推荐文章于 2024-09-02 07:40:27 发布

Zino-齐诺

最新推荐文章于 2024-09-02 07:40:27 发布

阅读量117

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_41938758/article/details/90041088

版权

leetcode 专栏收录该内容

71 篇文章 1 订阅

订阅专栏

在这里插入图片描述

/**
 * 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:
    void visit(TreeNode* root,int &u,int &a){
        if(!root) 
        {
            u=a=0;
            return;
        }
        int u1,u2,a1,a2;
        visit(root->left,u1,a1);
        visit(root->right,u2,a2);
        a=root->val+u1+u2;
        u=max(a1,u1)+max(a2,u2);
        return;
    }
    int rob(TreeNode* root) {
        int u,a;
        visit(root,u,a);
        return max(u,a);
    }
};