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

原题链接：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