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