就是一个树形动态规划,实现直接相连的节点之间不能都偷,然后找到如何偷到最大金额。
比如说现在有爷爷,两个孩子,四个孙子,那么爷爷需要的最多的就是可能就是爷爷的钱加四个孙子的钱或者是两个儿子的钱的和,儿子和孙子的钱通过递归来算,这样其实最终仍然是通过递归从底往上上的。
不如直接从底向上来动态规划。如果使用的是数组,则可以直接存储从小开始算,但树是没法存的,所以用一个map来存储每个节点对应的最大金额,这样在算的过程中就不用重复算了。
然后换一个思路可以和数组一样选择每个节点偷不偷,如果偷了,则儿子都选择不偷的最大值,如果不偷,则两个儿子选择偷的最大值。用一个两个长度的数组来表示偷或者不偷,0表示不偷,1表示偷,然后其中的值就是最大值,选其中更大的一个。
class Solution {
public int rob(TreeNode root) {
if(root==null)return 0;
int[] res=core(root);
return Math.max(res[0],res[1]);
}
public int[] core(TreeNode root){
if(root==null)return new int[]{0,0};
int[] res=new int[2];
int[] right=core(root.right);
int[] left=core(root.left);
res[0]=Math.max(right[0],right[1])+Math.max(left[1],left[0]);
res[1]=left[0]+right[0]+root.val;
return res;
}
}