leetcode 337. House Robber III-动态规划|Java|Python简洁高效

最新推荐文章于 2023-05-25 11:43:47 发布

Code_Granker

最新推荐文章于 2023-05-25 11:43:47 发布

阅读量7.2k

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/happyaaaaaaaaaaa/article/details/50880121

LeetCode 专栏收录该内容

167 篇文章

订阅专栏

本文详细解析了动态规划问题HouseRobberIII的递归实现方法，提供了Java和Python代码示例。通过递归函数dfs，实现了从叶子节点到指定节点的最大抢劫金额计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接：337. House Robber III

【思路-Java、Python】-递归实现

和 House Robber 十分类似，是Dynamic Programming问题。我们先考虑这样一个问题：

给定到从叶子节点到第 i - 2 层抢劫的最大值 max[i-2]，从叶子节点到第 i - 1 层抢劫的最大值 max[i-1]，第 i 层节点的值，那么是否可以得到从叶子节点到第 i 层节点的最大值呢？答案是肯定的。max[i] = Math.max(max[i-2]+val(a), max[i-1])。

这里使用递归来实现，数组rob来存储。rob[1]存储的是从叶子节点到当前节点抢劫的最大值，rob[0]存储的是从叶子节点到当前节点的左右孩子层节点抢劫到的最大值：

public class Solution {
    public int rob(TreeNode root) {
        return dfs(root)[1];
    }
    private int[] dfs(TreeNode root) {
        int[] rob ={0, 0};
        if(root != null) {
            int[] robLeft = dfs(root.left);
            int[] robRight = dfs(root.right);
            rob[0] = robLeft[1] + robRight[1];
            rob[1] = Math.max(robLeft[0] + robRight[0] + root.val, rob[0]);
        }
        return rob;
    }
}

124 / 124 test cases passed. Runtime: 1 ms

class Solution(object):
    def rob(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        def dfs(root) :
            if not root : return [0, 0]
            robLeft = dfs(root.left)
            robRight = dfs(root.right)
            norobCur = robLeft[1] + robRight[1]
            robCur = max(robLeft[0] + robRight[0] + root.val, norobCur)
            return [norobCur, robCur]
        return dfs(root)[1]

124 / 124 test cases passed. Runtime: 96 ms