标签：TreeNode val int notStealMax ReturnType 打家劫舍 root leetcode

题目描述

• 小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 root 。
• 除了 root 之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ，房屋将自动报警。
• 给定二叉树的 root 。返回 在不触动警报的情况下 ，小偷能够盗取的最高金额 。

链接：https://leetcode-cn.com/problems/house-robber-iii

使用二叉树递归套路

分析：该题通过考虑偷不偷根节点来考虑

• 第一种情况 ：
  偷根节点 ： 收益 = root.val + 在不偷左子树的最大收益 + 在不偷右子树的最大收益
• 第二种情况：
  注意： 在不偷根节点时，左右子树可以偷也可以不偷
  不偷根节点 : 收益 = 0 + max(偷左子树的最大收益, 不偷左子树的最大收益) + max(偷右子树的最大收益 , 不偷右子树的最大收益)
• ==> 返回值类型为 ：
  偷当前节点的最大收益（stealMax）, 不偷当前节点的最大收益（notStealMax

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    class ReturnType{
        int stealMax;
        int notStealMax;

        public ReturnType() {
        }

        public ReturnType(int stealMax, int notStealMax) {
            this.stealMax = stealMax;
            this.notStealMax = notStealMax;
        }
    }


    public int rob(TreeNode root) {
        if (root == null) {
            return 0;
        }
        
        // 只有一个节点时只能偷了
        if (root.left == null && root.right == null) {
            return root.val;
        }

        ReturnType result = process(root);

        return Math.max(result.notStealMax, result.stealMax);

    }

    private ReturnType process(TreeNode root) {
        if (root == null) {
            return new ReturnType(0,0);
        }
        
        ReturnType leftReturn = process(root.left);
        ReturnType rightReturn = process(root.right);

        // 偷root
        int stealRoot = root.val + leftReturn.notStealMax + rightReturn.notStealMax;
        // 不偷root
        int notStealRoot = Math.max(leftReturn.notStealMax, leftReturn.stealMax) + Math.max(rightReturn.notStealMax, rightReturn.stealMax);

        return new ReturnType(stealRoot,notStealRoot);

    }
}

标签：TreeNode,val,int,notStealMax,ReturnType,打家劫舍,root,leetcode
来源： https://blog.csdn.net/qq_43142792/article/details/123622467

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。