这道题一开始的确不好转化到tree的处理,这个学习了,最重要的是下面的错误,同上一题II,从根节点出发,这个无法变化,但若是从子节点出发,就完全可以是从子节点(这时相当于根)出发,或者,从子节点的子节点出发,所以必须要比较大小
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public int rob(TreeNode root) {
if (root == null) {
return 0;
}
int[] result = robHelper(root);
return Math.max(result[0], result[1]);
}
private int[] robHelper(TreeNode node) {
if (node == null) {
return new int[]{0, 0};
}
int[] left = robHelper(node.left);
int[] right = robHelper(node.right);
int[] result = new int[2];
result[0] = node.val + left[1] + right[1];
//result[1] = left[0] + right[0];
result[1] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
return result;
}
}